有沒有辦法在pandas的單列下對多列進行分組？

Question

我有一張這樣的桌子---

區域	列1	列2	列3
達卡	[1,2,3]	[2,3]	[4,5,6]
錫爾赫特	[10,11]	[7,9,10]	[10,13,7]

我想要這張桌子的形狀——

區域	總數
達卡	1個
	2個
	3個
	4個
	5個
	6個
錫爾赫特	10
	11
	7
	9
	13

我怎樣才能在 pandas 中做到這一點？

Answer 1

如有必要，將值轉換為列表：

import ast

df.iloc[:, 1:] = df.iloc[:, 1:].applymap(ast.literal_eval)

然后通過DataFrame.set_index和DataFrame.stack進行整形，通過Series.explode分解列表，創建 2 列 DataFrame 最后刪除重復項：

df = (df.set_index('AREA')
        .stack()
        .explode()
        .droplevel(1)
        .reset_index(name='total_num')
        .drop_duplicates())
print (df)
      AREA total_num
0    Dhaka         1
1    Dhaka         2
2    Dhaka         3
5    Dhaka         4
6    Dhaka         5
7    Dhaka         6
8   Sylhet        10
9   Sylhet        11
10  Sylhet         7
11  Sylhet         9
14  Sylhet        13