如何高效地读取CSV文件中的特定列？

使用 csv 模块选择性读取 CSV 文件中的列

理解问题

从 CSV 文件中读取特定列可能是一项常见任务在数据处理中。但是，按照给定代码中的尝试按列号访问列可能会导致意外结果。

解决错误

所提供代码中的错误在于将 print 语句放置在for 循环。因此，仅打印最后一次迭代的结果，即 CSV 文件的最后一列。

代码更正

要解决此问题，应将 print 语句移动到循环内进行打印每行的具体列：

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Pandas 模块的优点

而调整后的代码可以完成任务，缺乏pandas模块的便捷性和效率。 Pandas 为 CSV 文件操作提供了一种优雅且多功能的解决方案。

用于选择性列读取的 Pandas

使用 pandas，从 CSV 文件中读取特定列变得毫不费力：

import pandas as pd

df = pd.read_csv(csv_file)

# Save column data to a variable
column_data = df['column_name']

简化的代码

Pandas 解决方案显着简化了代码：

import pandas as pd

df = pd.read_csv(csv_file)

# Save specific columns as a list
specific_columns = ['ID', 'Name', 'Zip', 'Phone']
data = df[specific_columns].values.tolist()

以上是如何高效地读取CSV文件中的特定列？的详细内容。更多信息请关注PHP中文网其他相关文章！