板上高手好,
我目前想用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次?
感謝大家~