#python #pandas #dataframe #for-loop #counter
Вопрос:
Мне нужно реализовать счетчик, который выполняет подсчет, как показано на ВЫВОДЕ ниже. Он проверяет прошлые значения столбца «данные» на наличие отрицательных значений.
data output
0 -1 Nan // since there are no past values for data: count=NaN
1 -2 1 //-1, so count= 1
2 4 2 //-2,-1 count=2
3 12 0 // count=0
4 -22 0 // count=0
5 -12 1 //-22 count=1
6 -7 2 // -22,-12 count=2
7 -5 3 // -7,-22,-12 count=3
8 -33 4 // -5,7,-22,-12 count=4
9 2 5 // -33,-5,7,-22,-12 count=5
10 2 1 // count=0
МОЙ КОД
import pandas as pd
import talib
import numpy as np
df=pd.DataFrame()
df["data"]=[-1,-2,4,12,-22,-12,-7,-5,-33,2,2]
print(df)
c=0
for y in [0,len(ff)-1] :
for z in [1,10]:
if (ff["data"].shift(-z)).any()<=0:c=c 1
else:c
if (ff["data"].shift(-z)).any()>0:break
count["dd"]=c
Необходимые ВЫХОДНЫЕ данные:
Комментарии:
1. Вы уверены, что хотите получить такой результат? В строке 10 не должно ли быть написано 0?
Ответ №1:
Я довольно не уверен, как написать «Nan» (сам не очень хорош), но вот код, который, кажется, делает то, о чем вы просили:
df = pd.DataFrame()
df["data"] = [-1, -2, 4, 12, -22, -12, -7, -5, -22, 2, 2]
def generateOutput(df):
a = [0]
for i in range(len(df) - 1):
if df["data"][i] < 0:
a.append(a[-1] 1)
else:
a.append(0)
df["output"] = a
return df
print(df)
df = generateOutput(df)
print(df)
А вот мой вывод при запуске программы
data
0 -1
1 -2
2 4
3 12
4 -22
5 -12
6 -7
7 -5
8 -22
9 2
10 2
data output
0 -1 0
1 -2 1
2 4 2
3 12 0
4 -22 0
5 -12 1
6 -7 2
7 -5 3
8 -22 4
9 2 5
10 2 0
Комментарии:
1. Часть NaN не так уж и необходима. 0 было бы прекрасно.