np.array强制内部元素格式统一
  X5zJxoD00Cah 2023年11月02日 29 0
df_test = pd.DataFrame(
    data=np.array([
        np.repeat(['class' + str(i+1) for i in range(2)], 10),
        list(range(10, 0, -1)) * 2,
        list(range(1, 11)) * 2
    ]).T,
    columns=['class'] + ['x'+str(i+1) for i in range(2)]
)  # np.array强制内部元素格式统一

df_test[df_test.columns.drop(['class'])] = df_test[df_test.columns.drop(['class'])].astype('int32')

df_test['y'] = df_test.apply(
    func=lambda x: 10 + np.exp(x['x1']) + np.exp(2 * x['x2']) + 10
        if x['class'] == 'class1'
        else (
            20 + np.exp(10 * x['x1']) + np.exp(-2 * x['x2']) - 10
            if x['class'] == 'class2'
            else np.nan
        ),
    axis=1
)

df_test
【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

X5zJxoD00Cah