将数组范围作为参数传递给函数?

时间:2015-02-22 00:12:07

标签: python arrays numpy arguments

有没有办法将array-range作为参数传递给函数? 类似的东西:

> blah(ary,arg1=1:5)

def blah(ary,arg1): print ary[arg1]

4 个答案:

答案 0 :(得分:6)

您可以使用slice功能

>>> def blah(ary,arg1):
...     print ary[arg1]
>>> blah(range(10), slice(1, 5))
[1, 2, 3, 4]

答案 1 :(得分:6)

Python仅在方括号内接受1:5语法。解释器将其转换为slice对象。然后,对象的__getitem__方法应用切片。

查看numpy/lib/index_tricks.py以获取利用此功能的一些功能。实际上它们不是函数,而是定义自己的__getitem__方法的类。该文件可能会给你一些想法。

但如果你不能做到这一点,那么可能包括:

blah(arr, slice(1, 5))
blah(arr, np.r_[1:5])

nd_gridmgridogrid扩展'slice'概念以接受虚构的'step'值:

mgrid[-1:1:5j]
# array([-1. , -0.5,  0. ,  0.5,  1. ])

请注意,在将切片传递给blah函数之前在切片上展开的任何内容都不会知道另一个参数的形状。因此np.r_[:-1]只返回[]

None可以使用slice:例如slice(None,None,-1)相当于[::-1]

答案 2 :(得分:0)

你可以这样试试:

def blah(ary, arg):
    arg = map(int, arg.split(":"))
    print ary[arg[0]:arg[1]]

blah([1,2,3,4,5,6],"2:5")

输出:

[3, 4, 5]

答案 3 :(得分:0)

saw it happen today只是我很好奇,请注意,他们通过了 argument test_idx 作为纯范围

plot_decision_regions(X=X_combined_std, y=y_combined,
                      classifier=ppn, test_idx=range(105, 150))

那这到底是做什么的?

使用它切片 Numpy数组

ndarray [range(105,150),:]

但是,当我通过复制ndarray值进行测试时,实例化该值并尝试自己对其进行切片(基本上是创建一个列表),这将不允许我在该切片上传递该范围。

  

[[0.73088538 1.57698181],[0.17316034 0.1348488]

     

当您单击以设置新值时我从ndarray提取/复制的

     
    
      

erick = [[0.73088538 1.57698181],[0.17316034 0.1348488]]

    
  
     

-SyntaxError。语法无效

     

必须将逗号接受为一种语法并将其实例化为列表obj

     
    
      

erick = [[0.73088538,1.57698181],[0.17316034,0.1348488]

             

erick [:]

    
  
     

(可行,返回全部内容)

     
    
      

erick [range(0,1),:]

    
  
     

-TypeError。列表索引必须是整数或切片,而不是元组

     

(打破,我们之前测试过切片是否有效,所以它与范围有关)

enter image description here

尝试使用自制的Numpy Array(扰流板,可以正常工作)

erickNpa = np.asarray(erick, dtype=np.float32)
erickNpa[range(0, 1), :]

enter image description here


结论

您可以将范围作为参数传递,如第一部分所示,在某些时候,代码没有执行,但是必须可以根据正在执行的操作(使用它来分割列表)进行操作,但是如果使用正确的脚手架,一切似乎都可以正常工作,而使用Numpy Array可以证明这一点。



示例代码

我也将放 function def ,以防万一我连接了线路时git掉了。

def plot_decision_regions(X, y, classifier, test_idx=None, resolution=0.02):

    # setup marker generator and color map
    markers = ('s', 'x', 'o', '^', 'v')
    colors = ('red', 'blue', 'lightgreen', 'gray', 'cyan')
    cmap = ListedColormap(colors[:len(np.unique(y))])

    # plot the decision surface
    x1_min, x1_max = X[:, 0].min() - 1, X[:, 0].max() + 1
    x2_min, x2_max = X[:, 1].min() - 1, X[:, 1].max() + 1
    xx1, xx2 = np.meshgrid(np.arange(x1_min, x1_max, resolution),
                           np.arange(x2_min, x2_max, resolution))
    Z = classifier.predict(np.array([xx1.ravel(), xx2.ravel()]).T)
    Z = Z.reshape(xx1.shape)
    plt.contourf(xx1, xx2, Z, alpha=0.3, cmap=cmap)
    plt.xlim(xx1.min(), xx1.max())
    plt.ylim(xx2.min(), xx2.max())

    for idx, cl in enumerate(np.unique(y)):
        plt.scatter(x=X[y == cl, 0], 
                    y=X[y == cl, 1],
                    alpha=0.8, 
                    c=colors[idx],
                    marker=markers[idx], 
                    label=cl, 
                    edgecolor='black')

    # highlight test examples
    if test_idx:
        # plot all examples
        X_test, y_test = X[test_idx, :], y[test_idx]

        plt.scatter(X_test[:, 0],
                    X_test[:, 1],
                    c='',
                    edgecolor='black',
                    alpha=1.0,
                    linewidth=1,
                    marker='o',
                    s=100, 
                    label='test set')


# Training a perceptron model using the standardized training data:



X_combined_std = np.vstack((X_train_std, X_test_std))
y_combined = np.hstack((y_train, y_test))

plot_decision_regions(X=X_combined_std, y=y_combined,
                      classifier=ppn, test_idx=range(105, 150))
plt.xlabel('petal length [standardized]')
plt.ylabel('petal width [standardized]')
plt.legend(loc='upper left')

plt.tight_layout()
#plt.savefig('images/03_01.png', dpi=300)
plt.show()