usecols如何實(shí)現(xiàn)列的重命名

小樊
85
2024-06-14 11:30:32

要實(shí)現(xiàn)列的重命名，可以結(jié)合使用pandas庫(kù)的read_csv()函數(shù)的參數(shù)usecols和rename()函數(shù)。首先使用usecols參數(shù)選擇要讀取的列，然后使用rename()函數(shù)重命名這些列。

示例代碼如下：

import pandas as pd

# 讀取CSV文件并選擇要讀取的列
data = pd.read_csv('data.csv', usecols=['old_name1', 'old_name2', 'old_name3'])

# 重命名列
data = data.rename(columns={'old_name1': 'new_name1', 'old_name2': 'new_name2', 'old_name3': 'new_name3'})

# 打印重命名后的數(shù)據(jù)
print(data)

在上面的示例代碼中，首先使用usecols參數(shù)選擇了要讀取的列，然后使用rename()函數(shù)將這些列重命名為新的列名。

0 踩