板上高手好,
我目前想用numpy的frompyfunc对np.array进行广播
但是我是要对“每一列”进行计算(不是对每一列的每个值)
例如我要让我输入的引数是每一列假设3个数,然后输出一个list
def test_fn(a,b,c):
return a+1, b+2, c+5
input_array = np.array([[1,3,5],
[2,6,8]])
output = np.frompyfunc(test_fn, 3, 1)
然后我这边好像只能用这样写法:
outvalue = output(input_array[:, 0], input_array[:, 1], input_array[:, 2])
也就是我需要分别将a,b,c三个引数分别输入,
但我想知道有没有写法是我只要输入input_array变量就可以的?
不然如果一列50个引数我不就要输入50次?
感谢大家~
作者: fiorentino (眠) 2020-11-29 14:20:00
def test_fn(a):return a[0]+1, a[1]+2, a[2]+5outvalue = output(input_array)