返回一個列表，其中包含 dataframe 列值，這些值是根據另一個列表排序的

Question

我有一個帶有列ah的 df，我希望創建這些列值的列表，但按照另一個列表 ( list1 ) 中值的順序。 list1對應df中的索引值。

df

a  b  c  d  e  f  g  h

list1
[3,1,0,5,2,7,4,6]

想要的清單

['d', 'b', 'a', 'f', 'c', 'h', 'e', 'g']

Answer 1

你可以做df.columns[list1] ：

import pandas as pd
df = pd.DataFrame([], columns=list('abcdefgh'))

list1 = [3,1,0,5,2,7,4,6]

print(df.columns[list1])
# Index(['d', 'b', 'a', 'f', 'c', 'h', 'e', 'g'], dtype='object')

Answer 2

首先得到一個字母表的np.array

arr = np.array(list('abcdefgh'))

或者在您的情況下，您的df列的列表

arr = np.array(df.columns)

然后使用您的索引作為索引掩碼

arr[[3,1,0]]

出去：

['d', 'b', 'a']

Answer 3

查看

df.columns.to_series()[list1].tolist()