python - 将两个变量解压到 DataFrame 中的两列中

Question

我有一个名为 ReturnDateRange 的函数，它将返回两个日期。 我试图将此函数应用于数据帧中名为“zRow”的列，并将结果存储在两个不同的列中。

下面将把两个结果存储为一列中的元组:

df['t1']= df['zRow'].map(ReturnDateRange)

以下内容返回 ValueError:解压值太多(预期为 2)

df['t1'], df['t2']= df['zRow'].map(ReturnDateRange)

但该函数始终返回两个日期，或一个 None。

更新:我尝试返回两个零而不是“无”。仍然遇到同样的错误。

感谢您的帮助。

Answer 1

“太多值无法解包”错误是因为它解包的是行，而不是列。因此转置会有所帮助，但我觉得有更好的方法来解决这个问题。

同时，这可以工作，即使不是很优雅......

import pandas as pd
import numpy as np

data = pd.DataFrame(np.zeros(shape=(5,2)),columns=["a","b"])

def mapper(x):
    return ('first', 'second')

data['t1'], data['t2'] = data['b'].map(mapper).apply(pd.Series).values.T

print data

给出这个结果:

     a    b     t1      t2
0  0.0  0.0  first  second
1  0.0  0.0  first  second
2  0.0  0.0  first  second
3  0.0  0.0  first  second
4  0.0  0.0  first  second

这至少可以帮助其他人确定更好的解决方案。

归功于this也在这里发帖。

编辑，找到了更好的修复方法。使用:

data[['t1', 't2']] = data['b'].map(mapper).apply(pd.Series)

因此，就您而言，这应该有效:

df[['t1', 't2']] = df['zRow'].map(ReturnDateRange).apply(pd.Series)