python - 字符串向量列表Python

Question

我正在使用 Python 工作，数据框中有一列，它是一个字符串，看起来像这样:

df['set'] 

0  [911,3040]
1  [130055, 99832, 62131]
2  [19397, 3987, 5330, 14781]
3  [76514, 70178, 70301, 76545]
4  [79185, 38367, 131155, 79433]

我希望它是:

['911','3040'],['130055','99832','62131'],['19397','3987','5330','14781'],['76514',70178','70301','76545'],['79185','38367','131155','79433']

为了能够运行 Word2Vec:

model = gensim.models.Word2Vec(df['set'] , size=100)

谢谢!

Answer 1

如果您有一列字符串，我建议您查看 here以不同的方式解析它。

这是我的做法，使用 ast.literal_eval。

>>> import ast
>>> [list(map(str, x)) for x in df['set'].apply(ast.literal_eval)]

或者，使用pd.eval -

>>> [list(map(str, x)) for x in df['set'].apply(pd.eval)]  # 100 rows or less

或者，使用yaml.load -

>>> import yaml
>>> [list(map(str, x)) for x in df['set'].apply(yaml.load)]

[
     ['911', '3040'], 
     ['130055', '99832', '62131'], 
     ['19397', '3987', '5330', '14781'], 
     ['76514', '70178', '70301', '76545'],
     ['79185', '38367', '131155', '79433']
 ]