Python Pandas 获取 CSV 文件指定列数据

Python Pandas 获取 CSV 文件指定列数据

使用 Python Pandas 库，可以轻松地从 CSV 文件中获取指定列的数据。以下步骤和代码示例将详细说明如何实现。

1. 导入 Pandas 库

首先，需要导入 Pandas 库。

import pandas as pd

2. 读取 CSV 文件

使用 read_csv 函数读取 CSV 文件，并将其存储到一个 Pandas DataFrame 中。

df = pd.read_csv('filename.csv')

3. 获取指定列数据

假设我们要获取 CSV 文件中的第二列数据，可以使用以下代码：

col2 = df.iloc[:, 1]

其中，iloc 函数用于按行和列的索引位置进行访问，[:, 1] 表示获取所有行的第二列数据。

完整示例

import pandas as pd

df = pd.read_csv('filename.csv')
col2 = df.iloc[:, 1]
print(col2)

以上代码将打印 CSV 文件中第二列的所有数据。

总结

通过 Pandas 库，可以方便地从 CSV 文件中获取指定列数据。iloc 函数是进行索引访问的关键工具，可以根据行和列的索引位置选择需要的数据。