1. 程式人生 > >python陣列或列表刪除或讀取某一列

python陣列或列表刪除或讀取某一列

>>> import numpy as np
>>> data=[[1,2,3],[2,3,4],[4,5,6]]
>>> data
[[1, 2, 3], [2, 3, 4], [4, 5, 6]]
>>> #列表的讀取或刪除某一列
>>> #1.讀取
>>> data1=np.array(data)
>>> data1
array([[1, 2, 3],
       [2, 3, 4],
       [4, 5, 6]])
>>> data1[:,0]
array([1, 2, 4])
>>> #2.刪除
>>> np.delete(data1,0,axis=1)
array([[2, 3],
       [3, 4],
       [5, 6]])
>>> data1
array([[1, 2, 3],
       [2, 3, 4],
       [4, 5, 6]])
>>> #numpy陣列的讀取或刪除某一列
>>> data=np.array([[1,2,3],[2,3,4],[4,5,6]])
>>> data
array([[1, 2, 3],
       [2, 3, 4],
       [4, 5, 6]])
>>> #1.讀取
>>> data[:,1]
array([2, 3, 5])
>>> #2.刪除
>>> np.delete(data,1,axis=1)
array([[1, 3],
       [2, 4],
       [4, 6]])
>>> data
array([[1, 2, 3],
       [2, 3, 4],
       [4, 5, 6]])