使用Pandas读取CSV文件的特定列

  • Post category:Python

当我们需要从CSV文件中读取特定列的数据时,可以使用Pandas库中的read_csv()函数。下面是使用 Pandas 读取 CSV 文件中特定列数据的步骤。

步骤一:导入 Pandas 库

导入 Pandas 库,使用以下代码:

import pandas as pd

步骤二:指定 CSV 文件路径

使用以下代码指定要读取的 CSV 文件路径:

file_path = "path_to_csv_file.csv"

步骤三:使用 read_csv() 函数读取 CSV 文件

使用 read_csv() 函数读取 CSV 文件,并指定要读取的特定列,使用以下代码:

df = pd.read_csv(file_path, usecols=["column_name_1", "column_name_2", ...])

其中,usecols参数用于指定要读取的列名,可以直接传入列名列表,也可以传入列索引列表进行读取。

下面是一个具体的读取示例代码:

import pandas as pd

file_path = "data.csv"  # 指定 CSV 文件路径

# 指定要读取的列名
columns_to_load = ["Name", "Age"]

df = pd.read_csv(file_path, usecols=columns_to_load)  # 使用 read_csv 函数读取 CSV 文件

print(df)  # 打印 DataFrame

在此示例中,我们从 data.csv 文件中读取名字和年龄两列数据,然后打印输出了读取的 DataFrame 数据。

上述代码可以适用于读取任意 CSV 文件中的特定列数据,只需要修改文件路径和需要读取的列名即可。