Python if 函数跨越向量

Question

我是 Python 新手，正在尝试弄清楚如何执行以下操作。

我有一个数据数组 X，它看起来像:

X = [ 1 2 3

      4 5 6 

      7 8 9 

      10 11 12];

这样第一列X[0]是，

X[0] =[1
       4
       7
       10];

第二列X[1]是，

X[1] =[2
       5
       8
       11];

第三列X[2]是，

X[2] =[3
       6
       9
       12];

我想要做的是生成一个随机数k，并且如果第一列X[0]中的任何值大于k 我想将第三列乘以一个函数 (2*X[0])，如果任何值小于 k 我想要乘以不同的函数 (2*k)。

因此，如果k=2，则第三列变为:第三列 X[2] 是，

X[2] =[3*2*k
       6*2*X[0,1]
       9*2*X[0,2]
       12*2*X[0,3] ];

有没有一种简洁的方法可以做到这一点？

Answer 1

# inverted your rows and columns and got them in a list of list
X = [[1,4,7,10],
     [2,5,8,11],
     [3,6,9,12]]

# lets generate random number k between 0 and 1000
import random
k = random.randint(0, 1000)

# the conditional
for n in X[0]:
    if n > k:
        X[2] = your_function(X[2])
    if n < k:
        X[2] = your_other_function(X[2])

我很确定有更好的方法可以做到这一点，但这似乎很简单。当然 your_function 和 your_other_function 取决于你，无论如何我都不太清楚你想要什么。

还有，你考虑清楚了吗？如果 X[0] 中的所有值都完全等于 k，则这两个函数不会被应用的唯一方法是，这似乎不太可能。