Python 循环遍历数据框

Question

我有一个由股票价格组成的数据框。 对于一年，每个 TICKER、DATE、PRICE 是 1 行。 因此，每个代码有 365 行。 结构有点像这样：

    TICKER  PRICE_DATE  OPEN_PRICE
0      IBM    20201113      115.19
1      IBM    20201112      115.63
2      IBM    20201111      118.12
3      IBM    20201110      116.69
4      IBM    20201109      117.88
.
.
.
#      AAPL   20201113      115.19
#      AAPL   20201112      115.63
#      AAPL   20201111      118.12
#      AAPL   20201110      116.69
#      AAPL   20201109      117.88

我需要做的是：对于每个代码，循环前 3 天，并将开盘价与前一天进行比较。 基本上是这样的：IF day1 < day2 < day3 THEN........

所以，我需要 3 行 IBM，下一次迭代我需要 3 行 AAPL，下一次迭代......

有什么简单的方法可以做到这一点？ 或者，也许我应该将其转换为不同的格式，以便我可以访问 AAPL、价格元素 1、价格元素 2、价格元素 3 等？

Answer 1

数据.csv：

TICKER  PRICE_DATE  OPEN_PRICE
IBM    20201113      115.19
IBM    20201112      115.63
IBM    20201111      118.12
IBM    20201110      116.69
IBM    20201109      117.88
AAPL   20201113      115.19
AAPL   20201112      115.63
AAPL   20201111      118.12
AAPL   20201110      116.69
AAPL   20201109      117.88
INTC   20201113      115.19
INTC   20201112      115.63
INTC   20201111      118.12
INTC   20201110      116.69
INTC   20201109      117.88

代码：

import pandas as pd
import numpy as np

df = pd.read_csv('data.csv', delimiter='\s+')
print(df, '\n')

arr_t = np.dtype([('TICKER','S4')]) 
arr_t = df['TICKER'].unique()

df = df.set_index('TICKER')
print(df)
rows_per_ticker = df.loc[arr_t[0]].PRICE_DATE.count()

r = 0
while r < rows_per_ticker:
    for i in range(0,len(arr_t)):
        print(df.loc[arr_t[i]].iloc[r:r+3, :])
    r+=3

输出：

   TICKER  PRICE_DATE  OPEN_PRICE
0     IBM    20201113      115.19
1     IBM    20201112      115.63
2     IBM    20201111      118.12
3     IBM    20201110      116.69
4     IBM    20201109      117.88
5    AAPL    20201113      115.19
6    AAPL    20201112      115.63
7    AAPL    20201111      118.12
8    AAPL    20201110      116.69
9    AAPL    20201109      117.88
10   INTC    20201113      115.19
11   INTC    20201112      115.63
12   INTC    20201111      118.12
13   INTC    20201110      116.69
14   INTC    20201109      117.88

        PRICE_DATE  OPEN_PRICE
TICKER
IBM       20201113      115.19
IBM       20201112      115.63
IBM       20201111      118.12
IBM       20201110      116.69
IBM       20201109      117.88
AAPL      20201113      115.19
AAPL      20201112      115.63
AAPL      20201111      118.12
AAPL      20201110      116.69
AAPL      20201109      117.88
INTC      20201113      115.19
INTC      20201112      115.63
INTC      20201111      118.12
INTC      20201110      116.69
INTC      20201109      117.88
        PRICE_DATE  OPEN_PRICE
TICKER
IBM       20201113      115.19
IBM       20201112      115.63
IBM       20201111      118.12
        PRICE_DATE  OPEN_PRICE
TICKER
AAPL      20201113      115.19
AAPL      20201112      115.63
AAPL      20201111      118.12
        PRICE_DATE  OPEN_PRICE
TICKER
INTC      20201113      115.19
INTC      20201112      115.63
INTC      20201111      118.12
        PRICE_DATE  OPEN_PRICE
TICKER
IBM       20201110      116.69
IBM       20201109      117.88
        PRICE_DATE  OPEN_PRICE
TICKER
AAPL      20201110      116.69
AAPL      20201109      117.88
        PRICE_DATE  OPEN_PRICE
TICKER
INTC      20201110      116.69
INTC      20201109      117.88