Numpy中一维二维数组创建和单层双层for循环

# -*- coding: UTF-8 -*- 

import numpy as np

a = np.array([1, 2, 3])
print type(a)
sizea =  a.shape
print 'sizea: ', sizea

for i in range(sizea[0]):
    print 'a[', i, '] =', a[i]

b = np.array([[1,2,3],[4,5,6]])
c = np.array([[6,5,4],[1,2,3]])
sizeb = b.shape
print 'sizeb: ', sizeb

d = np.zeros((sizeb[0], sizeb[1]))  # Create an array of all zeros

for i in range( sizeb[0] ):
    for j in range( sizeb[1] ):
        d[i][j] = b[i][j] + c[i][j]

print '\nd= ', d

print 'Good bye'

运行结果：

<type 'numpy.ndarray'>
sizea:  (3L,)
a[ 0 ] = 1
a[ 1 ] = 2
a[ 2 ] = 3
sizeb:  (2L, 3L)

d=  [[7. 7. 7.]
 [5. 7. 9.]]
Good bye