Question

我正在尝试从连续的datetime变量中创建分类变量。

基本上，如果介于15到25之间，则Age_category = 1；如果介于25到35之间，则Age_category = 2。

我已经知道了，所以我可以提出一个条件，这可行：

df.loc[(df['Age_days'] > timedelta(days=(25*365)), 'Age2')] = '1'

当我尝试创建两个条件时，它会失败：

df.loc[(df['Age_days'] > timedelta(days=(15*365)) & (df['Age_days'] <= timedelta(days=(25*365)), 'Age_cat'))] = '1'

我得到的错误是它不喜欢使用'＆'：

TypeError: unsupported operand type(s) for &: 'datetime.timedelta' and 'tuple'

我尝试用＆替换＆，但它也不喜欢这样：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有什么主意我应该解决这个问题吗？

Answer 1

对于Pandas，您应该使用Pandas方法而不是datetime模块。

在这种情况下，您可以使用NumPy将timedelta的值转换为float，然后使用pd.cut来区分年龄。默认情况下，pd.cut包含在每个范围的右侧。

df = pd.DataFrame({'Age': [10, 15, 17, 20, 25, 30, 35, 40]})
df['Age'] = pd.to_timedelta(df['Age'], unit='Y')

df['Category'] = pd.cut(df['Age'] / np.timedelta64(1, 'Y'), [15, 25, 35], labels=[1, 2])

print(df)

                  Age Category
0  3652 days 10:12:00      NaN
1  5478 days 15:18:00      NaN
2  6209 days 02:56:24      1.0
3  7304 days 20:24:00      1.0
4  9131 days 01:30:00      1.0
5 10957 days 06:36:00      2.0
6 12783 days 11:42:00      2.0
7 14609 days 16:48:00      NaN

然后，您可以删除空值并将类别转换为int（如果需要）。

从连续的日期时间变量中生成分类变量

1 个答案: