1. 程式人生 > >Python建立二維陣列和切片操作

Python建立二維陣列和切片操作

在推薦系統的資料構造時,經常要使用二維陣列,通常先建立元素均為0的陣列,這要使用Numpy模組中的zeros()函式

1.建立一個元素為0的二維陣列:

from numpy import *
b=zeros((2,3))
>>
array([[ 0.,  0.,  0.],
       [ 0.,  0.,  0.]])
2.給陣列賦不同值,將第1行賦值為1,2,3
a[0]=[1,2,3]
>>
array([[ 1.,  2.,  3.],
       [ 0.,  0.,  0.]])
3.分片操作,選取二維陣列的第一列,第二列
a[:,0]
>>
array([ 1.,  0.])
a[:,1]
>>
array([ 2.,  0.])
4.分片操作,選取二維陣列的第一行,第二行
a[0]
>>
array([ 1.,  2.,  3.])
a[1]
>>
array([ 0.,  0.,  0.])