Python - 如何選擇 Pandas DataFrame 的一個子集


假設 Microsoft Excel 中開啟的 CSV 檔案的內容如下 −

首先,將 CSV 檔案中的資料載入到 Pandas DataFrame 中 −

dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")

要選擇一個子集,使用方括號。在括號中提及列並從整個資料集中提取單列 −

dataFrame['Car']

示例

以下是程式碼 −

import pandas as pd

# Load data from a CSV file into a Pandas DataFrame
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")
print("\nReading the CSV file...\n",dataFrame)

# displaying only a single column
res1 = dataFrame['Car'];
# displaying only a subset
print("\nDisplaying only one column Car : \n",res1)

# displaying two columns
res2 = dataFrame[['Car','Units']];
# displaying another subset
print("\nDisplaying two columns : \n",res2)

輸出

這將產生以下輸出 −

Reading the CSV file...
       Car   Reg_Price   Units
0      BMW        2500     100
1    Lexus        3500      80
2     Audi        2500     120
3   Jaguar        2000      70
4  Mustang        2500     110

Displaying only one column Car :
0      BMW
1    Lexus
2     Audi
3   Jaguar
4  Mustang
Name: Car, dtype: object

Displaying two columns :
       Car   Units
0      BMW     100
1    Lexus      80
2     Audi     120
3   Jaguar      70
4  Mustang     110

更新於: 29-Sep-2021

324 次瀏覽

事業起步

透過完成課程獲得認證

入門
廣告
© . All rights reserved.