如果尺寸匹配,则列表或numpy数组可以解包为多个变量。对于3xN阵列,将执行以下操作:
import numpy as np
a,b = [[1,2,3],[4,5,6]]
a,b = np.array([[1,2,3],[4,5,6]])
# result: a=[1,2,3], b=[4,5,6]
Run Code Online (Sandbox Code Playgroud)
如何对熊猫的柱子实现类似的行为DataFrame?扩展以上示例:
import pandas as pd
df = pd.DataFrame([[1,2,3],[4,5,6]])
df.columns = ['A','B','C'] # Rename cols and
df.index = ['i', 'ii'] # rows for clarity
Run Code Online (Sandbox Code Playgroud)
以下内容无法正常工作:
a,b = df.T
# result: a='i', b='ii'
a,b,c = df
# result: a='A', b='B', c='C'
Run Code Online (Sandbox Code Playgroud)
但是,我想得到的是以下内容:
a,b,c = unpack(df)
result: a=df['A'], b=df['B'], c=df['C']
Run Code Online (Sandbox Code Playgroud)
该功能unpack已经在熊猫中可用吗?还是可以通过简单的方式模仿它?