在python中，通过使用lambda将空列表添加到dataframe列会引发valueError

Question

我能够在综合数据上重现该错误：

import pandas as pd

from datetime import datetime

df1 = pd.DataFrame({'A': ['A0', 'A1', 'A2', 'A3'],

                    'B': ['B0', 'B1', 'B2', 'B3'],

                    'C': [datetime.now(), datetime.now(), datetime.now(), datetime.now()],

                    'D': ['D0', 'D1', 'D2', 'D3']},

                   index=[0, 1, 2, 3]);

df2 = pd.DataFrame({'A': ['A1', 'A2', 'A3', 'A4'],

                    'E': ['E1', 'E2', 'E3', 'E4']},

                   index=[0,1,2,3]);

df = pd.merge(df1, df2, how='left', on=['A', 'A']);

def getList(row):

    r = [];

    if row["A"] == "A1": r.append("test-01");

    if row["B"] == "B1": r.append("test-02");

    if row["B"] == "B2": r.append("test-03");

    return r;

df["NEW_COLUMN"] = df.apply(lambda row: getList(row), axis = 1);

原始帖子：我想基于多种条件在pandas数据框中创建一个新列。新列的值应为list。但是我收到“ ValueError：指定索引传递的空数据。” 如果列表为空。

def getList(p_row):

  r = [];

  if p_row["field1"] > 0: r.append("x");

  ...

  return r;

df["new_list_field"] = df.apply(lambda row: getList(row), axis = 1);

完整的错误：

ValueError追溯（最近一次通话最近）C：\ ProgramData \ Anaconda3 \ lib \ site-packages \ pandas \ core \ internals.py在create_block_manager_from_arrays（数组，名称，轴）4636中尝试：-> 4637块= form_blocks（数组，名称） ，轴）4638 mgr = BlockManager（块，轴）

C：\ ProgramData \ Anaconda3 \ lib \ site-packages \ pandas \ core \ internals.py in form_blocks（数组，名称，轴）4728如果len（object_items）> 0：-> 4729 object_blocks = _simple_blockify（object_items，np.object_ ）4730 blocks.extend（object_blocks）

C：\ ProgramData \ Anaconda3 \ lib \ site-packages \ pandas \ core \ internals.py in _simple_blockify（tuples，dtype）4758“”“-> 4759值，位置= _stack_arrays（tuples，dtype）4760

C：\ ProgramData \ Anaconda3 \ lib \ site-packages \ pandas \ core \ internals.py in _stack_arrays（tuples，dtype）4822 for i，enr in enumerate（arrays）：-> 4823 Stacked [i] = _asarray_compat（arr） 4824

ValueError：无法将输入数组从形状（2）广播到形状（195）

Answer 1

最终制作了一个列表列表，将其转换为，pd.Series()然后将其分配给新列。字典key2list返回可变长度列表作为值：new_col_list = []for _, row in my_df.iterrows():    new_col_list.append(key2list[row[u'key']])my_df[u'new_col'] = pd.Series(new_col_list)

Answer 2

该函数的输出长度因行而异，但是您不能将不等长的列表分配给新的pandas列。您可以通过以下方式进行验证：for idx,row in df.iterrows():    print(getList(row))一种替代方法是将输出转换为字符串：def getListString(row):    r = ''    if row["A"] == "A1": r+="test-01"    if row["B"] == "B1": r+="test-02"    if row["B"] == "B2": r+="test-03"    return r