python - 即使函数适用于每个列系列，带str.extract的DataFrame.apply也会引发错误

使用此示例数据帧：df = pd.DataFrame([['A-3', 'B-4'], ['C-box', 'D1-go']])
在各个列上调用extract作为序列工作正常：

df.iloc[:, 0].str.extract('-(.+)')
df.iloc[:, 1].str.extract('-(.+)')

并且在另一个轴上：

df.iloc[0, :].str.extract('-(.+)')
df.iloc[1, :].str.extract('-(.+)')

所以，我希望使用apply可以工作（通过对每个列应用摘录）：
df.apply(lambda s: s.str.extract('-(.+)'), axis=0)
但它抛出了这个错误：

Traceback (most recent call last):
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\IPython\core\interactiveshell.py", line 3325, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-588-70b1808d5457>", line 2, in <module>
    df.apply(lambda s: s.str.extract('-(.+)'))
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\frame.py", line 6487, in apply
    return op.get_result()
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\apply.py", line 151, in get_result
    return self.apply_standard()
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\apply.py", line 260, in apply_standard
    return self.wrap_results()
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\apply.py", line 308, in wrap_results
    return self.wrap_results_for_axis()
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\apply.py", line 340, in wrap_results_for_axis
    result = self.obj._constructor(data=results)
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\frame.py", line 392, in __init__
    mgr = init_dict(data, index, columns, dtype=dtype)
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\internals\construction.py", line 212, in init_dict
    return arrays_to_mgr(arrays, data_names, index, columns, dtype=dtype)
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\internals\construction.py", line 51, in arrays_to_mgr
    index = extract_index(arrays)
  File "C:\ProgramData\Miniconda3\envs\py3\lib\site-packages\pandas\core\internals\construction.py", line 308, in extract_index
    raise ValueError('If using all scalar values, you must pass'
ValueError: If using all scalar values, you must pass an index

使用axis=1会产生一个意外结果，其中每一行都是一个序列：

Out[2]:
0             0
0  3
1  4
1         0
0  box
1   go
dtype: object

我使用apply，因为我认为这将导致最快的执行时间，但是可以接受其他建议

最佳答案

str.extract中expand的默认参数为True，它返回一个数据帧。由于要将其应用于多个列，因此它会尝试返回多个数据帧。设置expand为False来处理这个问题，

df.apply(lambda x: x.str.extract('-(.*)', expand = False))

    0   1
0   3   4
1   box go

关于python - 即使函数适用于每个列系列，带str.extract的DataFrame.apply也会引发错误，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/58794687/