如何在特定列中获取列表的并集

时间:2019-03-03 03:36:15

标签: python-3.x pandas

首先,我用熊猫读取excel文件,然后像这样df.groupby(“ wire”):

        wire    pin  app             re_range
W1  0    W1      1   K1                   [1]
    2    W1    2,3   K2                [2, 3]
    1    W1      4   K3                   [4]
    5    W1    7~9   K4             [7, 8, 9]
    4    W1  10,11   T4              [10, 11]
    3    W1  21~25   T3  [21, 22, 23, 24, 25]
W2  6    W2      1   K6                   [1]
W3  7    W3      1   K8                   [1]
W4  8    W4      1   K5                   [1]
W5  9    W5      1   K9                   [1]
W6  10   W6      1   K7                   [1]
W7  11   W7      1  K19                   [1]

因此,我如何才能将W1中的所有re_range列表合并在一起,例如: [1,2,3,4,7,8,9,10,11,21,22,23,24,25]

我尝试使用“ lambda x”方法,但似乎不起作用。

2 个答案:

答案 0 :(得分:1)

使用

df.re_range.sum(level=0)

答案 1 :(得分:0)

也许您可以不用分组来做到:

w1_df = df[df['wire']=='w1']
ranges = w1_df['re_range']
all_ranges = []
for range in ranges:
    all_ranges.extend(range)