在 Python 中将 csv 文件特定列提取到列表

若要在 Python 中将 csv 文件中的特定列提取到列表，我们可以使用 Pandas read_csv() 方法。

步骤

• 列出要提取的列。

• 使用 read_csv() 方法将 csv 文件提取到数据框中。

• 打印提取的数据。

• 使用 plot() 方法绘制数据框。

• 若要显示图形，请使用 show() 方法。

示例

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
columns = ["Name", "Marks"]
df = pd.read_csv("input.csv", usecols=columns)
print("Contents in csv file:\n", df)
plt.plot(df.Name, df.Marks)
plt.show()

csv 文件包含以下数据 −

输出

当我们执行该代码时，它将从 csv 文件中提取数据并显示以下图 −

Rishikesh Kumar Rishi

更新于： 2021 年 5 月 7 日

1.5 万+ 次浏览

启动您的 职业生涯

通过完成课程获得认证

开始学习