提取数组元素并组成新的数组

不知道N4错哪了
import numpy as np
def return_values():
import step1
N4=step1.return_values()
N5=np.array([N4[0][1],N4[0][3],N4[2][0],N4[2][4]])

NameError: name 'N4' is not defined
重新定义N4
输出N5