Блог

Я хотел бы добавить столбец с повторяющимися значениями ряда в фрейм данных, при этом каждое значение в списке повторяется определенное количество раз, например:

    x   y  z
   0   2  1
   2  15  1
   0  10  2
   2  20  2
   0  30  3
   2  50  3
  

Где каждое значение в z повторяется дважды.

Пока у меня есть это, которое повторяет весь список, а не каждый элемент в списке заданное количество раз:

 data = [[0, 2], [2, 15], [0, 10], [2, 20], [0, 30], [2, 50]]
df1 = pd.DataFrame(data, columns = ['x', 'y'])

df1['z']=0
np.put(df1['z'], np.arange(len(df1)), [1, 2])
print(df1)
  

Результат:

 x   y  z
0   2  1
2  15  2
0  10  3
2  20  1
0  30  2
2  50  3
  

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

Не уверен, отвечает ли это на ваш вопрос, но:

использование понимания списка

 ntimes = 2 # number of times you repeat each item
yourlist = [1,2,3]
df1['z'] = [i for i in yourlist for _ in range(ntimes)]
  

делает то, что вы ищете.

    x   y  z
0  0   2  1
1  2  15  1
2  0  10  2
3  2  20  2
4  0  30  3
5  2  50  3