重新启动数据框中的循环

Question

我有此数据框，

   index turns conv
0      0  utt1  yes
1      1  utt2  yes
2      2  utt3   no
3      3  utt4  yes
4      0  utt5  yes
5      1  utt6   no
6      2  utt7  yes

我想打印“转弯”列的两个元素和“ conv”列的相应元素，但在索引0上重新启动了loop 的，以便utt4和utt5 don t连接。我拥有的代码是：

for i in range(len(df['turns'])):
    if(i+1==len(df['turns'])):
      break;
    else:
      print(df['turns'][i], df['turns'][i+1], df['conv'][i+1]) 

但是当前输出：

utt1 utt2 yes
utt2 utt3 no
utt3 utt4 yes
utt4 utt5 yes
utt5 utt6 no
utt6 utt7 yes

虽然我需要它来输出：（

utt1 utt2 yes
utt2 utt3 no
utt3 utt4 yes

utt5 utt6 no
utt6 utt7 yes

这个想法是一个滑动窗口的想法，但我无法弄清楚如何以更简单的方式做到这一点）

Answer 1

如果您只想打印，则可以将循环更改为：

for i in range(len(df['turns'])-1):
    if df.loc[i+1, 'index'] == 0:
        print()
    else:
        print(df['turns'][i], df['turns'][i+1], df['conv'][i+1]) 

输出：

utt1 utt2 yes
utt2 utt3 no
utt3 utt4 yes

utt5 utt6 no
utt6 utt7 yes

矢量解决方案是：

group = df['index'].eq(0).cumsum()
(df
 .assign(turns2=df.groupby(group)['turns'].shift())
 .dropna(subset=['turns2'])
 [['turns2', 'turns', 'conv']]
 .to_csv('out.csv', index=False, header=False, sep=' ')
)

out.csv：

utt1 utt2 yes
utt2 utt3 no
utt3 utt4 yes
utt5 utt6 no
utt6 utt7 yes