Python Создание массива с помощью функции

Question

я пытался сделать эту матрицу

array([[0,1,2], 
     [10,11,12], 
     [100,101,102], 
     [110,111,112]]) 

с этой функцией

def f(x,y): 
    if x < 2: 
     return 10 * x + y 
    else : 
     return 100 + 10 * x + 6 

и

print(np.fromfunction(f,(4,3),dtype=int) 

Однако, это дало мне ошибку

The truth value of an array with more than one element is ambiguous. Use a.any() or a.all() 

Я думаю, x не означает точный ряд, который я хотел использовать.

Как я могу исправить эту функцию, чтобы получить результат выше, и могу ли я знать, в чем проблема?

Answer 1

import numpy as np 

def f(i,j): 
    return np.where(i<2, i*10+j, 100+j+10*(i-2) ) 

print np.fromfunction(f ,(4,3),dtype=int) 

выход:

[[ 0 1 2] 
[ 10 11 12] 
[100 101 102] 
[110 111 112]] 

np.where возвращается отдельный элемент, а не w дыра.

Answer 2

f,(4,3) clear comma after "f". 

Answer 3

Причина, по которой это не работает, заключается в том, что функция переходит к своим первым аргументам, а не к int.

код жалуется, потому что a < 2 является массив/INT сравнения, вы должны запустить a.all() < 2

Рабочий код становится:

from numpy import fromfunction 

def f(x,y): 
    if x.all() < 2: 
     return 10 * x + y 
    else : 
     return 100 + 10 * x + 6 

print fromfunction(f,(3,3),dtype=int)