如何交换 pandas 中的两列(附示例)
您可以使用以下自定义函数来交换 pandas DataFrame 中两列的位置:
def swap_columns (df, col1, col2): col_list = list ( df.columns ) x, y = col_list. index (col1), col_list. index (col2) col_list[y], col_list[x] = col_list[x], col_list[y] df = df[col_list] return df
此函数将交换 DataFrame 中列col1和col2的位置。
下面的例子展示了如何在实际中使用这个功能。
示例:交换 Pandas 中的两列
假设我们有以下 pandas DataFrame:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'], ' points ': [18, 22, 19, 14, 14, 11, 20, 28], ' assists ': [5, 7, 7, 9, 12, 9, 9, 4], ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]}) #view DataFrame print (df) team points assists rebounds 0 A 18 5 11 1 B 22 7 8 2 C 19 7 10 3 D 14 9 6 4 E 14 12 6 5 F 11 9 5 6 G 20 9 9 7:28 4 12
我们可以定义一个swap_columns()函数来交换“points”和“bounces”列的位置:
#define function to swap columns def swap_columns (df, col1, col2): col_list = list ( df.columns ) x, y = col_list. index (col1), col_list. index (col2) col_list[y], col_list[x] = col_list[x], col_list[y] df = df[col_list] return df #swap points and rebounds columns df = swap_columns (df, ' points ', ' rebounds '): #view updated DataFrame print (df) team rebounds assists points 0 A 11 5 18 1 B 8 7 22 2 C 10 7 19 3 D 6 9 14 4 E 6 12 14 5 F 5 9 11 6 G 9 9 20 7:12 a.m. 4:28
请注意,“得分”和“篮板”列已交换,而其他所有列仍保持在相同位置。
其他资源
以下教程解释了如何在 pandas 中执行其他常见操作:
Pandas:如何计算列中特定值的出现次数
Pandas:获取列与值匹配的行的索引
Pandas:如何计算DataFrame中的缺失值