Python 当输入是字典时，`pandas. concat`如何工作

发布于03月18日

我很难理解，当输入是一本词典时，pd.concat是如何工作的.

假设我们有以下Pandas 数据帧-

# Import pandas library
import pandas as pd

# initialize list of lists
data = [['tom', 10], ['nick', 15], ['juli', 14]]

# Create the pandas DataFrame
df = pd.DataFrame(data, columns=['Name', 'Age'])

然后，我们进行以下连接操作—

z = pd.concat({"z":df},
              axis = 1)

print(z)

输出结果是—

      z    
   Name Age
0   tom  10
1  nick  15
2  juli  14

看起来密钥z被堆叠在数据帧df的顶部.但这没有意义，因为指定的轴是1，因此，堆叠(如果发生了这种情况)应该是跨列的.

推荐答案

这实际上是有意义的，因为你连接为columns(axis=1)，你需要区分连接的列.

下面是一个更有意义的例子:

out = pd.concat({'left': df.add_prefix('left_'),
                 'middle': df.add_prefix('middle_'),
                 'right': df.add_prefix('right_')},
                axis=1)

       left               middle                 right          
  left_Name left_Age middle_Name middle_Age right_Name right_Age
0       tom       10         tom         10        tom        10
1      nick       15        nick         15       nick        15
2      juli       14        juli         14       juli        14

这相当于将新名称传递给keys:

out = pd.concat([df.add_prefix('left_'),
                 df.add_prefix('middle_'),
                 df.add_prefix('right_')],
                keys=['left', 'middle', 'right'],
                axis=1)

如果连接axis=0(行)，则concat将作为索引级的前缀:

out = pd.concat({'top': df,
                 'middle': df,
                 'bottom': df},
                 axis=0)

          Name  Age
top    0   tom   10
       1  nick   15
       2  juli   14
middle 0   tom   10
       1  nick   15
       2  juli   14
bottom 0   tom   10
       1  nick   15
       2  juli   14