Python 使用 pandas Select 包含存储在 set 列中的特定 int 的行

发布于09月21日

我试着在网上搜索，我放弃了……我想请求您在以下方面提供帮助(我使用的是Python3和Pandas):

假设df的Values列包含集合中的整数:

在:

            VALUES
0           {2624, 1383, 2855, 1871, 16784, 9811, 2970}
1           {2898, 12, 1871, 223}
2           {2624, 2855, 519, 15662, 1230, 1871, 2970}
3          {2624, 12, 55, 519, 313, 1230, 10588, 2970}
4           {1230, 55, 89564, 1247}

Out Desired(仅包含包含特定值的行，例如，值集中的12行，其他所有行都将丢弃):

            VALUES
1           {2898, 12, 1871, 223}
3          {2624, 12, 55, 519, 313, 1230, 10588, 2970}

在下一步中，我将需要将int12移到新的单独列.此外，我也找不到一种方法来转换列表中的这些集合.

               VALUES                                  NEW_COL
1           [2898, 1871, 223]                             12
3          [2624, 55, 519, 313, 1230, 10588, 2970]        12

谢谢.

推荐答案

匹配项筛选:

booleans = [12 in val for val in df.VALUES]
out = df.loc[booleans]

从值中删除12以获得最终输出:

out.assign(VALUES = out.VALUES - {12}, NEW_COL = 12)

                                    VALUES  NEW_COL
1                        {2898, 1871, 223}       12
3  {2624, 519, 1230, 55, 313, 2970, 10588}       12

如果要转换为列表:

out.assign(VALUES = (out.VALUES - {12}).map(list), NEW_COL = 12)
                                    VALUES  NEW_COL
1                        [2898, 1871, 223]       12
3  [2624, 519, 1230, 55, 313, 2970, 10588]       12

我不确定您在做什么，但在性能方面，最好是处理诸如列表/集/元组之类的 struct ……在Pandas 外面