Как я могу создать новый столбец с различными наборами значений в зависимости от индекса?

Question

Я хочу создать новые столбцы для DataFrame, которые будут содержать значение в зависимости от набора индекса

Вот DataFrame:

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

Вот что я хочу получить:

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

Я хочу создать новый столбец на основе набора индекса.

Answer 1

У меня есть набор данных футбола, я хочу добавить круглую колонку; для каждых 10 совпадений я хочу присвоить одно значение.

Используйте целочисленное деление, если по умолчанию RangeIndex на 10, и добавьте 1:

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

Или, если необходимо общее решение создать вспомогательный массив:

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