Панды: повторить n раз значение каждой из n строк

Question

У меня есть датафрейм с 2563199 строками.Это выглядит так:

          index    dtm        f
      0     0   00:00:00    50.065
      1     1   00:00:01    50.061
      2     2   00:00:02    50.058
      3     3   00:00:03    50.049
      4     4   00:00:04    50.044
      5     5   00:00:05    50.044
      6     6   00:00:06    50.042
      7     7   00:00:07    50.042
    ....................
   2591997  2591997 23:59:57    50.009
   2591998  2591998 23:59:58    50.008
   2591999  2591999 23:59:59    50.006

Я хотел бы создать новый столбец, который повторяет значение, содержащееся в каждом n строках n раз.Например, если я установлю значение повтора в строке 4, оно будет повторять 50.049 в предыдущих 4 строках, а значение 50.042 в следующих 4 и т. Д.(Если длина кадра данных не совпадает с точным делением, не имеет значения).Как показано ниже:

          index   dtm         f
      0     0   00:00:00    50.049
      1     1   00:00:01    50.049
      2     2   00:00:02    50.049          
      3     3   00:00:03    50.049
      4     4   00:00:04    50.042
      5     5   00:00:05    50.042
      6     6   00:00:06    50.042
      7     7   00:00:07    50.042

Я пробовал каждые 86400 строк:

arr = np.arange(len(df)) // 86400
for x in arr:
    df['value']=df['f'].iloc[x+86400]

Есть идеи?спасибо!

Answer 1

Вот способ сделать это, избегая зацикливания на df.

. Сначала установите n и создайте список с существующими индексами, исключая строки, которые будут использоваться для повторения значений вf:

n=4
ix = [x for i, x in enumerate(df.index.values) if (i + 1) % n != 0]
print(ix)
[0, 1, 2, 4, 5, 6]

Теперь установите эти значения на np.nan и используйте bfill:

df.loc[ix, 'f'] = np.nan
df['f'] = df.f.bfill()

print(df)
    index       dtm       f
0      0  00:00:00  50.049
1      1  00:00:01  50.049
2      2  00:00:02  50.049
3      3  00:00:03  50.049
4      4  00:00:04  50.042
5      5  00:00:05  50.042
6      6  00:00:06  50.042
7      7  00:00:07  50.042

Answer 2

Использование numpy и разбиение массива

import numpy as np

n = 4
df['fnew'] = np.concatenate([np.repeat(df.f.values[n-1::n], n),
                             np.repeat(np.NaN, len(df)%n)])

Вывод:

n=3
   index       dtm       f    fnew
0      0  00:00:00  50.065  50.058
1      1  00:00:01  50.061  50.058
2      2  00:00:02  50.058  50.058
3      3  00:00:03  50.049  50.044
4      4  00:00:04  50.044  50.044
5      5  00:00:05  50.044  50.044
6      6  00:00:06  50.042     NaN
7      7  00:00:07  50.042     NaN

n = 4
   index       dtm       f    fnew
0      0  00:00:00  50.065  50.049
1      1  00:00:01  50.061  50.049
2      2  00:00:02  50.058  50.049
3      3  00:00:03  50.049  50.049
4      4  00:00:04  50.044  50.042
5      5  00:00:05  50.044  50.042
6      6  00:00:06  50.042  50.042
7      7  00:00:07  50.042  50.042

n = 5
   index       dtm       f    fnew
0      0  00:00:00  50.065  50.044
1      1  00:00:01  50.061  50.044
2      2  00:00:02  50.058  50.044
3      3  00:00:03  50.049  50.044
4      4  00:00:04  50.044  50.044
5      5  00:00:05  50.044     NaN
6      6  00:00:06  50.042     NaN
7      7  00:00:07  50.042     NaN