我可以使用数组切片而不是创建向量吗?

时间:2011-09-18 19:55:33

标签: python numpy slice

这不是功课! 只是复杂问题的例子。 现在我采用数组,创建向量,然后使用带向量的方程作为输入。 一切正常。但是,想知道我是否可以跳过矢量创建并使用数组切片到同一个东西。

import numpy as np

a = [[ 7.,  5.], [ 9.,  3.], [ 7.,  2.], [ 7.,  8.], [ 6.,  8.], [ 3.,  2.]]
a = np.asarray(list(a), np.float)

x = np.asarray(a[:,(0)], np.float)
y = np.asarray(a[:,(1)], np.float)

foo = 5*x+y*y 
print foo 

1 个答案:

答案 0 :(得分:5)

import numpy as np

a = np.array([[ 7.,  5.], [ 9.,  3.], [ 7.,  2.], [ 7.,  8.], [ 6.,  8.], [ 3.,  2.]])

foo = 5*a[:,0] + a[:,1]**2