我有这样一个数据帧:
df = pd.DataFrame(index=['pre1_xyz', 'pre1_foo', 'pre3_bar', 'pre3_foo', 'pre10_foo', 'pre10_bar', 'pre10_xyz'])
我想添加一个列values
,其中的值是根据使用函数return_something(pref)
的相应行的索引前缀来确定的。现在我实现如下:
然后得到期望的输出:
values
pre1_xyz 0
pre1_foo 0
pre3_bar 7
pre3_foo 7
pre10_foo 13
pre10_bar 13
pre10_xyz 13
问题是是否有比这更聪明的解决方案,例如避免创建unique_pref
和/或{
因为前缀有重复,所以需要首先分离出前缀,以确保不会为同一前缀生成新的随机数。因此,必须从前缀列表中删除重复项。我以一种更简洁的方式完成了这项工作,为前缀创建了一个新列,然后使用df.prefix.unique(). 在
相关问题 更多 >
编程相关推荐