熊猫将系列分配给分组结果

Question

嗨，我对如何将系列分配给分组结果有点一无所知。

我有数据帧 A 和 B：

A = pd.DataFrame({'ID':[1,1,1,2,2,2],'TW':[0,1,0,0,1,0]})
B = pd.DataFrame({1:['A','B','C'], 2:['A','B','C']})

B 的列表示 A 的 ID 信息。我想按 ID 对 A 进行分组，并分配 B 的相应列来替换 TW 数据。这是我想要的：

C = pd.DataFrame({'Date':[1,1,1,2,2,2],'TW':['A','B','C','A','B','C']})

有人可以帮忙吗？

Answer 1

你不能融化B吗？

>>> pd.melt(B, var_name='Date', value_name='TW')
Date TW
0     1  A
1     1  B
2     1  C
3     2  A
4     2  B
5     2  C

Answer 2

@Alexander的答案是显而易见的。但这是另一种方法。

B.rename_axis('Date', 1).stack() 
.reset_index('Date', name='TW') 
.reset_index(drop=True)