python - 在 pandas DataFrame 多索引函数之外解压列表

Question

我想将多索引列添加到现有的 pandas 数据框 df。一个例子:

d = {('a','b'):[1,2,3], ('c', 'd'): [4,5,6]}
df = pd.DataFrame(d)

生成的数据框是:

   a  c 
   b  d 
0  1  4 
1  2  5 
2  3  6 

现在我想向数据框中添加一个新列。正确的方法是使用 df['e', 'f'] = [7,8,9]。但是，我想使用列表 new_key 作为键。通常我可以使用星号*，但显然它不能在函数之外使用。所以我得到以下错误。

new_key = ['e','f']

df[new_key] = [7,8,9]
> KeyError: "['e' 'f'] not in index"

df[*new_key] = [7,8,9]
> SyntaxError: invalid syntax 

有人知道怎么解决吗？

Answer 1

首先转换为元组:

df[tuple(new_key)] = [7,8,9]

   a  c  e
   b  d  f
0  1  4  7
1  2  5  8
2  3  6  9