如何使用 Python 从 CSV 文件中提取特定列？

使用 Python csv 模块修剪 CSV 列

通常，需要解析逗号分隔值 (CSV) 文件并提取特定的数据列。使用Python的csv模块，可以轻松完成这一任务。

列提取

要提取特定列，请使用included_cols变量，该变量将列索引指定为包括。每列从 0 开始编号，第一列为 0。例如，要捕获“ID”、“名称”、“邮政编码”和“电话”列，您可以编写 Included_cols = [1, 2, 6 , 7].

迭代和打印

要迭代行并提取指定列，请在 for 循环中使用列表理解。对于每一行，内容变量将使用指定列中的值进行填充。要显示提取的数据，请在循环中使用 print(content)。

included_cols = [1, 2, 6, 7]

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

使用 Pandas 进行优化

为了更高效地处理 CSV，请考虑使用 pandas 模块。 Pandas 提供了方便且强大的工具来读取和操作 CSV 文件。要读取 CSV 并将特定列存储到变量中，只需使用：

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

以上是如何使用 Python 从 CSV 文件中提取特定列？的详细内容。更多信息请关注PHP中文网其他相关文章！