在 Numpy/Pandas 中創建條件循環

Question

這里絕對是新手....

我有一個包含費用數據列表的數據集1我想創建一個循環來確定該人花費比前一天更多並且也比第二天花費更多的日期。 這樣做時，我希望它要么打印日期和金額（費用），要么創建一個新的列，閱讀真/假。 我應該使用 Numpy 還是 Pandas？

我在想這樣的事情：今天 = 我昨天 = i-1 明天 = i+1 ...然后繼續創建一個循環

Answer 1

你在尋找這樣的東西：

# sample data
np.random.seed(4)
df = pd.DataFrame({'Date': pd.date_range('2020-01-01', '2020-01-10'),
                   'Name': ['Some Name', 'Another Name']*5,
                   'Price': np.random.randint(100,1000, 10)})

# groupby name 
g = df.groupby('Name')['Price']

# create a mask to filter your dataframe where the current price is grater than the price above and below
mask = (g.shift(0) > g.shift(1)) & (g.shift(0) > g.shift(-1))
df[mask]

        Date          Name  Price
3 2020-01-04  Another Name    809
4 2020-01-05     Some Name    997
7 2020-01-08  Another Name    556