主页 / 问题 / 1224677
Accepted
Владислав Харламов
Asked:2021-12-28 05:27:58 +0800 CST2021-12-28 05:27:58 +0800 CST

从 SQL array_agg 转换数据

  • 772

我有一个 CSV 表，其中有一列包含这样的行：

    {сообщение один, сообщение два, сообщение три}    
    {NULL, NULL, сообщение пятнадцать}
...

我想阅读它，以便它变成这样的字符串列表：

['сообщение один', 'сообщение два', 'сообщение три']
['NULL', 'NULL', 'сообщение пятнадцать']

如何实施？

python

1 个回答

  1. Best Answer

    例子：

    In [355]: df = pd.read_csv("a.csv", sep="~", header=None, names=["text"])

In [356]: df
Out[356]: 
                                             text
0  {сообщение один, сообщение два, сообщение три}
1              {NULL, NULL, сообщение пятнадцать}

In [357]: df["res"] = df["text"].str.replace(r"[{}]", "").str.split(",\s*")

In [358]: df.loc[0, "res"]
Out[358]: ['сообщение один', 'сообщение два', 'сообщение три']

In [359]: df.loc[1, "res"]
Out[359]: ['NULL', 'NULL', 'сообщение пятнадцать']

In [360]: df
Out[360]: 
                                             text                                             res
0  {сообщение один, сообщение два, сообщение три}  [сообщение один, сообщение два, сообщение три]
1              {NULL, NULL, сообщение пятнадцать}              [NULL, NULL, сообщение пятнадцать]
    • 3

相关问题