python - 将多个字符串列转换为整数列表

Question

我有一个像这样的数据框 df，其中两列都是 object。

    +-----+--------------------+--------------------+
    |  id |         col1       |         col2       |
    +-----+--------------------+--------------------+
    |   1 |  0,1,4,0,1         |  1,2,4,0,0         |
    +-----+--------------------+--------------------+

我将它们转换成这样的列表

test = df["col1"]+','+df["col2"]
test.tolist()

作为列表中的 SINGLE STING 元素产生以下结果

['0,1,4,0,1,1,2,4,0,0']

但是，我希望它们像这样的整数列表

[0,1,4,0,1,1,2,4,0,0] 

有什么建议吗？仅供引用，在我的原始数据集中，这些列真的很大，所以性能也可能是一个问题。

Answer 1

我想你想要:

(df['col1'] + ',' + df['col2']).apply(lambda row: [int(s) for s in row.split(',')])

输出:

0    [0, 1, 4, 0, 1, 1, 2, 4, 0, 0]
dtype: object