values = [1,2,3,2,3,1]
colors = ['r','g','b']
expected_output = ['r', 'g', 'b', 'g', 'b', 'r'] # how to create this in pandas?
df = pd.DataFrame({'values': values})
df['colors'] = expected_output
即,我想在我的数据框中创建一个新列,其中 colored颜色 是基于现有列中的值 Select 的.我记得在XARRAY中用矢量化索引技巧做到了这一点,但我记不清同样的事情在Pandas 身上是否可能发生……感觉这应该是一个基本的索引任务吗?
编辑:目前的答案是一个很好的开始,谢谢!不过,他们有点太过利用"值"的数字本质.我宁愿一些通用的,也可以工作,如果说
values = ['a', 'b', 'c', 'b', 'c', 'a']
我想" map "的方法可能还有效.