pandas中提取DataFrame某些列的一些方法

2022-06-06 14:02:56

前言
方法一：df[columns]
方法二：df.loc[]：用 label （行名或列名）做索引。
方法三：df.iloc[]: i 表示 integer，用 integer location（行或列的整數位置，從0開始）做索引。
補充：提取所有列名中包含“線索”、“瀏覽”欄位的列
參考：
總結

前言

在處理表格型資料時，一行資料是一個 sample，列就是待提取的特徵。怎麼選取其中的一些列呢？本文分享一些方法。

使用如下的資料作為例子：

import pandas as pd

data = pd.DataFrame({'Name':['Anna', 'Betty', 'Richard', 'Philip','Paul'],
        'course1':[85,83,90,84,85],
        'course2':[90,85,83,88,84],
        'course3':[82,86,81,91,85],
        'fruit':['apple','banana','apple','orange','peach'],
        'sport':['basketball', 'volleyball', 'football', 'basketball','baseball']},
         index=[1,2,3,4,5])
  
df = pd.DataFrame(data)

df

	Name	course1	course2	course3	fruit	sport
1	Anna	85	90	82	apple	basketball
2	Betty	83	85	86	banana	volleyball
3	Richard	90	83	81	apple	football
4	Philip	84	88	91	orange	basketball
5	Paul	85	84	85	peach	baseball

方法一：df[columns]

先看最簡單的情況。輸入列名，選擇一列。例如：

df['course2']

1    90
2    85
3    83
4    88
5    84
Name: course2, dtype: int64

df[column list]：選擇列。例如：

df[['course2','fruit']]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

或者以 column list (list 變數)的形式匯入到 df[ ] 中，例如：

select_cols=['course2','fruit']
df[select_cols]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

可以用 column list=df.columns[start:end] 的方式選擇連續列，start 和 end 均為數位，不包括 end 列。例如：

select_cols=df.columns[1:4]
df[select_cols]

	course1	course2	course3
1	85	90	82
2	83	85	86
3	90	83	81
4	84	88	91
5	85	84	85

你可能注意到，其中有 3 列的名字相近：‘course1’,‘course2’,‘course3’。怎麼提取這三列呢？這裡分享在Kaggle 上看到一位大神使用的 list comprehension方法。

select_cols=[c for c in df.columns if 'course' in c]
df[select_cols]

	course1	course2	course3
1	85	90	82
2	83	85	86
3	90	83	81
4	84	88	91
5	85	84	85

但是，如果你想輸入df['course1':'course3'] 來索引連續列，就會報錯。而輸入數位索引df[1:3]時，結果不再是列索引，而是行索引，如下所示：

df[1:3]

	Name	course1	course2	course3	fruit	sport
2	Betty	83	85	86	banana	volleyball
3	Richard	90	83	81	apple	football

以下兩種方法 df.loc[]和df.iloc[]就可以解決這個問題，可以明確行或列索引。還可以同時取多行和多列。

方法二：df.loc[]：用 label （行名或列名）做索引。

輸入 column_list 選擇多列 [:, column_list]，括號中第一個: 表示選擇全部行。例如：

df.loc[:,['course2','fruit']]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

選擇連續多列 [:,start_col: end_col]，注意：包括 end_col。例如：

df.loc[:,'course2':'fruit']

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple
4	88	91	orange
5	84	85	peach

選擇多行和多列，例如：

df.loc[1:3,'course2':'fruit']

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple

與 df[ ]類似，df.loc[ ]括號內也可以輸入判斷語句，結果是對行做篩選。例如：

df.loc[df['course1']>84]
#注：輸入df[df['course1']>84]，輸出結果相同

	Name	course1	course2	course3	fruit	sport
1	Anna	85	90	82	apple	basketball
3	Richard	90	83	81	apple	football
5	Paul	85	84	85	peach	baseball

方法三：df.iloc[]: i 表示 integer，用 integer location（行或列的整數位置，從0開始）做索引。

df.iloc與df.loc用法類似，只是索引項不同。

df.iloc[:,[2,4]]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

選擇連續多列：df.iloc[:, start_ix:end_ix]，注意：不包括 end_ix。例如：

df.iloc[:,2:5]

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple
4	88	91	orange
5	84	85	peach

選擇多行與多列，例如：

df.iloc[1:3,[2,4]]

	course2	fruit
2	85	banana
3	83	apple

與 df.loc[] 不同，df.iloc[] 括號內不可以輸入判斷語句。

補充：提取所有列名中包含“線索”、“瀏覽”欄位的列

import pandas as pd
 
path = 'F:python_projectspython_learningershouche.csv'
df = pd.read_csv(open(path), index_col=0)
df = df.fillna(0)  # 填充空值後需賦值
print(df.describe())
 
columns = df.columns.values.tolist()  # 獲取列名列表，注意values，tolist的使用
col_xian = []  # 儲存包含‘線索'欄位的列名
for i in columns:
    if '線索' in i:
        col_xian.append(i)
 
col_liu = [] # 儲存包含‘瀏覽'欄位的列名
for i in columns:
    if '瀏覽' in i:
        col_liu.append(i)
 
df_xian = df[col_xian]  # 根據列名取列
df_liu = df[col_liu]

參考：

1.如何選取dataframe的多列-教學：https://www.geeksforgeeks.org/how-to-select-multiple-columns-in-a-pandas-dataframe/

2.用 list comprehension 選擇多列：https://www.kaggle.com/code/robikscube/ieee-fraud-detection-first-look-and-eda/notebook

3.df.loc 與 df.iloc 的比較：https://stackoverflow.com/questions/31593201/how-are-iloc-and-loc-different

總結

到此這篇關於pandas中提取DataFrame某些列的文章就介紹到這了,更多相關pandas提取DataFrame某些列內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

pandas中提取DataFrame某些列的一些方法

目錄

前言

方法一：df[columns]

方法二：df.loc[]：用 label （行名或列名）做索引。

方法三：df.iloc[]: i 表示 integer，用 integer location（行或列的整數位置，從0開始）做索引。

補充：提取所有列名中包含“線索”、“瀏覽”欄位的列

參考：

總結

熱門文章