如何使用 CSV 模块从 CSV 文件中的特定列提取数据？

使用 CSV 模块从 CSV 文件中的特定列提取数据

在处理 CSV 文件时，经常需要提取数据仅来自特定列。要使用 CSV 模块实现此目的，了解该库如何处理数据检索非常重要。

在提供的代码示例中，尝试通过 row[i] 语法访问特定列，其中 i 代表列号。然而，由于方法不正确，结果并没有达到预期。

正确的方法是将 print 语句包含在 for 循环中。以下更正后的代码演示了这一点：

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

现在，修改后的代码将打印出每行所需列中的值。

但是，值得一提的是使用以下方法的替代方法： pandas 库，简化了 CSV 文件的处理。使用 pandas，您可以轻松读取 CSV 文件并将特定列提取到变量中：

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

pandas 库提供了一组用于处理结构化数据的多功能工具，使其成为处理 CSV 文件的推荐选择有效。

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