【python】pandas:自定义列名

在Pandas中读取文件时，可以通过以下两种方式自定义列名，本文以读取CSV文件为例。

1. 在读取文件后自定义列名：

   • 首先，使用pd.read_csv()函数读取CSV文件，不指定列名（或者CSV文件本身就没有列名）。
   • 然后，直接给DataFrame的columns属性赋值新的列名列表。

   示例代码：

   import pandas as pd
   
   df = pd.read_csv('example.csv', header=None)  # 假设CSV文件没有列名
   df.columns = ['列名1', '列名2', '列名3', ...]  # 用列表定义新的列名
   

2. 在读取文件的同时自定义列名：

   • 使用pd.read_csv()函数读取CSV文件，并通过names参数指定新的列名列表。
   • 如果CSV文件本身有列名但你不想使用，可以添加header=None参数来忽略原有列名。

   示例代码：

   import pandas as pd
   
   df = pd.read_csv('example.csv',  names=['列名1', '列名2', '列名3', ...]) # 文件原本没有列名
   

   或者，如果CSV文件有列名但你想替换它们：

   import pandas as pd
   
   df = pd.read_csv('example_with_header.csv', header=None,names=['列名1', '列名2', '列名3', ...])  # 忽略原有列名，使用新的列名
   

注意：

• names参数中的列名列表长度应该与CSV文件中的列数相匹配。
• 如果CSV文件有列名但想忽略它们，并同时指定新的列名，可以使用header=None参数与names参数一起。
• 如果CSV文件没有列名，并且想使用默认的数字列名（如0, 1, 2,…），这时不需要指定header或names参数。pandas会自动生成列名0, 1, 2,…。