pandas - 将字符串转换为字符串列表

Question

我有这个'file.csv'文件用pandas读取:

Title|Tags
T1|"[Tag1,Tag2]"
T1|"[Tag1,Tag2,Tag3]"
T2|"[Tag3,Tag1]"

运用

df = pd.read_csv('file.csv', sep='|')

输出是:

  Title              Tags
0    T1       [Tag1,Tag2]
1    T1  [Tag1,Tag2,Tag3]
2    T2       [Tag3,Tag1]

我知道该列Tags是一个完整的字符串,因为:

In [64]: df['Tags'][0][0]
Out[64]: '['

我需要把它读作一个字符串列表["Tag1","Tag2"].我尝试了这个问题中提供的解决方案,但没有运气,因为我有[和]字符实际搞乱事情.

期望的输出应该是:

In [64]: df['Tags'][0][0]
Out[64]: 'Tag1'

Answer 1

您可以使用内置的ast.literal_eval，它适用于元组和列表

\n

import ast\nimport pandas as pd\n\ndf = pd.DataFrame({"mytuples": ["(1,2,3)"]})\n\nprint(df.iloc[0,0])\n# >> \'(1,2,3)\'\n\ndf["mytuples"] = df["mytuples"].apply(ast.literal_eval)\n\nprint(df.iloc[0,0])\n# >> (1,2,3)\n

\n\n

编辑：eval应该避免！如果正在评估的字符串是，os.system(\xe2\x80\x98rm -rf /\xe2\x80\x99)它将开始删除计算机上的所有文件（此处）。提供的ast.literal_eval字符串或节点可能仅包含以下 Python 文字结构：字符串、字节、数字、元组、列表、字典、集合、布尔值和 None（此处）。谢谢@TrentonMcKinney :)

\n

Answer 2

您可以手动拆分字符串：

>>> df['Tags'] = df.Tags.apply(lambda x: x[1:-1].split(','))
>>> df.Tags[0]
['Tag1', 'Tag2']

Answer 3

我认为您可以使用 json 模块。

import json
import pandas

df = pd.read_csv('file.csv', sep='|')
df['Tags'] = df['Tags'].apply(lambda x: json.loads(x))

因此，这将像以前一样加载您的数据框，然后将 lambda 函数应用于Tags列中的每个项目。lambda 函数调用json.loads()将列表的字符串表示形式转换为实际列表。

Answer 4

strip您可以使用和将字符串转换为列表split。

df_out = df.assign(Tags=df.Tags.str.strip('[]').str.split(','))

df_out.Tags[0][0]

输出：

'Tag1'

Answer 5

要么

df.Tags=df.Tags.str[1:-1].str.split(',').tolist()