python-3.x - 如何根据索引创建具有不同值集的新列？

Question

我想为 DataFrame 创建一个新列，该列将包含取决于索引集的值

这是数据框：

data = pd.DataFrame({'name': ['name1', 'name2', 'name3', 'name4'], 
                          } )

这是我想要得到的：

data = pd.DataFrame({'name': ['name1', 'name2', 'name3', 'name4'], 
                      'new_value': ['1', '1', '2', '2']
                      } )

我想根据索引集创建一个新列。

标签： python-3.xpandasnumpy

Answer 1

我有足球数据集，我想添加圆柱；对于每 10 场比赛，我想分配一个值。

如果默认使用整数除法RangeIndex并10添加1：

data['new_value'] = data.index // 10  + 1

或者如果需要通用解决方案创建辅助数组：

data['new_value'] = np.arange(len(data)) // 10 + 1