熊猫，减去列Dataframe in loop

Question

我是熊猫新手。 我有一个由6列组成的数据框，我想做一个for循环来做到这一点：

-创建一个新列（nc 1）-nc1 =列1-列2

我想对所有列进行迭代，所以最后一个将是：ncx =列5-列6

我可以减去这样的列：

df['nc'] = df.Column1 - df.Column2

但这在我尝试执行循环时没有用，因为我总是必须插入列的名称。

有人可以告诉我如何将列称为数字吗？ 谢谢！

Answer 1

In [26]: import numpy as np
    ...: import random
    ...: import pandas as pd
    ...: 
    ...: A = pd.DataFrame(np.random.randint(100, size=(5, 6)))

In [27]: A
Out[27]: 
    0   1   2   3   4   5
0  82  13  17  58  68  67
1  81  45  15  11  20  63
2   0  84  34  60  90  34
3  59  28  46  96  86  53
4  45  74  14  10   5  12

In [28]: for i in range(0, 5):
    ...:     A[(i + 6)] = A[i] - A[(i + 1)]
    ...: 
    ...: 
    ...: A
    ...: 
Out[28]: 
   0   1   2   3   4   5   6   7   8   9   10
0  82  13  17  58  68  67  69  -4 -41 -10   1
1  81  45  15  11  20  63  36  30   4  -9 -43
2   0  84  34  60  90  34 -84  50 -26 -30  56
3  59  28  46  96  86  53  31 -18 -50  10  33
4  45  74  14  10   5  12 -29  60   4   5  -7

In [29]: nc = 1 #The first new column
    ...: A[(nc + 5)] #outputs the first new column
Out[29]: 
0    69
1    36
2   -84
3    31
4   -29

在这里，您不必通过名称（仅通过列号）来调用它，而只需编写一个简单的函数即可调用列+ 5

像这样：

In [31]: def call_new_column(n):
    ...:     return(A[(n + 5)])
    ...: 
    ...: 
    ...: call_new_column(2)
Out[31]: 
0    -4
1    30
2    50
3   -18
4    60