Python Pandas Dataframe 存儲下一行值

Question

我目前是 python 數據幀的新手，並嘗試遍歷行。 我希望能夠獲取下 2 行的值並將其存儲在變量中。 以下是代碼片段：

df = pd.read_csv('Input.csv')
for index, row in df.iterrows():
    row_1 = row['Word']
    row_2 = row['Word'] + 1 # I know this is incorrect and it won't work
    row_3 = row['Word'] + 2 # I know this is incorrect and it won't work
    print(row_1, row_2, row_3)

我希望給定（'Input.csv'）：

Word <- #Column
 Hi
 Hello
 Some
 Phone
 keys
 motion
 entries

我想要 output 如下：

Hi, Hello, Some
Hello, Some, Phone
Some, Phone, keys
Phone, keys, motion
keys, motion, entries

任何幫助表示贊賞。 謝謝你。

Answer 1

有很多方法可以做到這一點，但像這樣使用 Pandas 矢量化操作的簡單方法將起作用，

(df['Word'] + ', ' + df['Word'].shift(-1)+ ', ' + df['Word'].shift(-2)).dropna()

Answer 2

你可以簡單地使用iloc屬性

df = pd.read_csv('Input.csv')
for index, row in df.iterrows():
    row_1 = df.iloc[index]['word']
    row_2 = df.iloc[index + 1]['word']
    row_3 = df.iloc[index + 2]['word'] 
    print(row_1, row_2, row_3)