pandas CSV输出在添加截断参数时会有 ['']

2024-03-29 11:31:15 发布

您现在位置:Python中文网/ 问答频道 /正文

我成功地添加了一个截止选项,以便在熊猫中获得接近的匹配。出于某种原因,当我添加cutoff=0.7时,当它输出到我的CSV时,它会读作['Name']。当它没有截止参数时,它只输出不带[''的匹配。下面是我的代码。任何帮助都将不胜感激!你知道吗

https://pastebin.com/nRt3rVPr

import sys
import difflib
import csv
import pandas as pd

df = pd.read_csv(sys.argv[1])
List = df.list #.astype(str).values.tolist()
MappedID = df.Party.astype(str)
df['Name_r'] = MappedID.map(lambda x: (difflib.get_close_matches(x, List, cutoff=0.7)[:1] or [None][0]))
header = [ "List", "Party", "Name_r"]
df.to_csv(sys.argv[2], columns = header)
print(df.to_string())

Tags: csvnameimportdfpartysyslistheader