选择配对行，Col Pandas值

Question

我敢肯定这很容易，但是发现或发现答案不是！我试图从DF（5,5）的行大小（5,5）的行中选择纬度（行：5,10,15等）和经度（COLS：-98，-97，-96等）的配对值）。假设我的DF看起来像这样：

DF：

  lat -98 -97 -96 -95 -94
0   5   6   7   8   9  10
1  10  11  12  13  14  15
2  15  16  17  18  19  20
3  20  21  22  23  24  25
4  25  26  27  28  29  30

要通过单行和单一col进行提取的对迭代，我需要以下内容：

dfnew:
0   6
1   12 
2   18
3   24
4   30

我在下面尝试了类似的事情，不同类型的循环太多了，无法在此处显示：

df.iloc[0:5,1:6] 

但是所有的行和所有列，我只需要单个配对值。 谢谢你，

Answer 1

iiuc，您可以使用numpy.diag：

import numpy as np
out = pd.Series(np.diag(df.drop(columns='lat')), index=df.index)

输出：

0     6
1    12
2    18
3    24
4    30

如果您想要一对长对，则可能：

out = pd.DataFrame({'lat': df['lat'], 'long': np.diag(df.drop(columns='lat'))})

输出：

   lat  long
0    5     6
1   10    12
2   15    18
3   20    24
4   25    30