在 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 萬+ 次瀏覽

啟動您的 職業生涯

透過完成課程獲得認證

開始學習