将熊猫系列(带字符串)转换为 python 列表

时间:2021-01-12 17:08:29

标签: python pandas series tolist

这可能是一件愚蠢的事情,但我似乎无法正确地将最初从 Excel 工作表获得的熊猫系列转换为列表。

dfCI 是通过从 Excel 工作表导入数据创建的,如下所示:

tab      var             val
MsrData  sortfield       DetailID
MsrData  strow           4
MsrData  inputneeded     "MeasDescriptionTest", "SiteLocTest", "SavingsCalcsProvided","BiMonthlyTest"     

# get list of cols for which input is needed
cols = dfCI[((dfCI['var'] == 'inputneeded') & (dfCI['tab'] == 'MsrData'))]['val'].values.tolist()
print(cols)

>> ['"MeasDescriptionTest", "SiteLocTest", "SavingsCalcsProvided", "BiMonthlyTest"']

# replace null text with text
invalid = 'Input Needed'
for col in cols:
   dfMSR[col] = np.where((dfMSR[col].isnull()), invalid, dfMSR[col])

但是,当我将 cols 从系列转换为列表时添加的第二组(单)引号使所有列成为单个值,以便

col = '"MeasDescriptionTest", "SiteLocTest", "SavingsCalcsProvided", "BiMonthlyTest"'

cols 的期望输出是

cols = ["MeasDescriptionTest", "SiteLocTest", "SavingsCalcsProvided", "BiMonthlyTest"]

我做错了什么?

2 个答案:

答案 0 :(得分:2)

获得 col 后,您可以将其转换为预期的输出:

In [1109]: col = '"MeasDescriptionTest", "SiteLocTest", "SavingsCalcsProvided", "BiMonthlyTest"'

In [1114]: cols = [i.strip() for i in col.replace('"', '').split(',')]

In [1115]: cols
Out[1115]: ['MeasDescriptionTest', 'SiteLocTest', 'SavingsCalcsProvided', 'BiMonthlyTest']

答案 1 :(得分:1)

考虑到 cols 的结构,我想到的另一个可能的解决方案是:

list(eval(cols[0]))  # ['MeasDescriptionTest', 'SiteLocTest', 'SavingsCalcsProvided', 'BiMonthlyTest']

虽然这是有效的,但它不太安全,我会按照@MayankPorwal 的建议使用列表理解。