腳本之家服務(wù)器常用軟件

快捷導航

Python 用NumPy創(chuàng)建二維數(shù)組的案例

更新時間：2021年03月11日 10:54:53 作者：捉貓的耗子

這篇文章主要介紹了Python 用NumPy創(chuàng)建二維數(shù)組的案例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

前言

上位機實戰(zhàn)開發(fā)先放一放，今天來學習一個新的內(nèi)容—NumPy的使用

1 一維數(shù)組

例：用普通方法生成一維數(shù)組

num = [0 for i in range(1,5)] # 創(chuàng)建一維數(shù)組
print(num) # 打印數(shù)組
print("-"*50) # 分割線
num[2]=6 # 將第三個元素修改位6
print(num) # 打印數(shù)組
print("-"*50) # 分割線

運行結(jié)果

例：用numpy生成一維數(shù)組

from numpy import *
m1 = arange(5)
print(m1)
print("-"*50) # 分割線

運行結(jié)果

2 二維數(shù)組

例：用普通方法生成二維數(shù)組

num = [[0 for i in range(1,5)]for j in range(1,5)]
print(num) # 打印數(shù)組
print("-"*50) # 分割線
num[2][1]=3 # 將第三行第二個元素修改為3
print(num) # 打印數(shù)組
print("-"*50) # 分割線

運行結(jié)果

例：用numpy生成二維數(shù)組

from numpy import *
m1 = array([arange(1,4),arange(4,7),arange(7,10),arange(10,13),arange(13,16),arange(16,19)]) # 創(chuàng)建6行3列的二維數(shù)組
print(m1)
print("-"*50) # 分割線

運行結(jié)果

通過上圖發(fā)現(xiàn)，用普通方法創(chuàng)建的數(shù)組分隔符有逗號，而numpy創(chuàng)建的數(shù)組分隔符默認為空格。

3 二維數(shù)組的分片

請看程序注釋以及結(jié)果對照

from numpy import *
m1 = array([arange(1,4),arange(4,7),arange(7,10),arange(10,13),arange(13,16),arange(16,19)]) # 創(chuàng)建6行3列的二維數(shù)組
print(m1)
print("-"*50) # 分割線
print(m1[1][0:]) # [x][0:] 獲取第x行的全部元素（單行）
print("-"*50) # 分割線
print(m1[1:3]) # [x1:x2] 獲取第x1~x2行的全部元素（多行）
print("-"*50) # 分割線
print(m1[1::4]) # [x1::x2] 分片操作，從第x1行開始，步長為x2

運行結(jié)果

4 數(shù)組維度的操作

.reshape：將一維數(shù)組變成多維數(shù)組

用法：一維數(shù)組名.reshape(x,y,z)將一維數(shù)組改成三維數(shù)組

from numpy import *
m1 = array([arange(1,9)]) # 創(chuàng)建一維數(shù)組
print(m1)
print("-"*50) # 分割線
print(m1.reshape(2,2,2)) # 將一維數(shù)組轉(zhuǎn)為三維數(shù)組
print("-"*50) # 分割線

運行結(jié)果

.ravel或flatten：將多維數(shù)組變?yōu)橐痪S數(shù)組

用法：數(shù)組名.ravel（）或數(shù)組名.flatten（）

from numpy import *
m1 = array([arange(1,4),arange(4,7),arange(7,10),arange(10,13),arange(13,16),arange(16,19)]) # 創(chuàng)建6行3列的二維數(shù)組
print(m1)
print("-"*50) # 分割線
print(m1.ravel()) # 將一維數(shù)組轉(zhuǎn)為三維數(shù)組
print("-"*50) # 分割線

運行結(jié)果

.shape：將高維數(shù)組轉(zhuǎn)換成低維數(shù)組

用法：數(shù)組名.shape（x行，列）

.transpose：將數(shù)組進行轉(zhuǎn)置

用法：數(shù)組名.transpose()

from numpy import *
m1 = array([arange(1,4),arange(4,7),arange(7,10),arange(10,13),arange(13,16),arange(16,19)]) # 創(chuàng)建6行3列的二維數(shù)組
print(m1)
print("-"*50) # 分割線
print(m1.transpose()) # 將數(shù)組轉(zhuǎn)置
print("-"*50) # 分割線

運行結(jié)果

5 數(shù)組的組合與分隔

5.1 水平組合

hastck：將數(shù)組水平拼接到一起

用法：hastck((數(shù)組A，數(shù)組B))水平拼接數(shù)組A和B

5.2 垂直組合

vstack:將數(shù)組垂直拼接到一起

用法：vstack(數(shù)組A，數(shù)組B)垂直拼接數(shù)組A和B

5.3 水平分隔

hsplit：

用法：hsplit（數(shù)組名，x）x為分成幾部分

5.4 垂直分隔

vsplit：

用法：vsplit（數(shù)組名，x）x為分成幾部分

6 數(shù)組轉(zhuǎn)列表

.tolist：將數(shù)組轉(zhuǎn)列表

用法：數(shù)組名.tolist()

7 數(shù)組的存儲和讀取及分隔符

savetxt和loadtxt函數(shù)可將數(shù)組保存為CSV文件。CSV文件的分隔符包括空格，逗號，分號等。savetxt和loadtxt函數(shù)默認使用的時空格作為分割符，也可以用delimiter關(guān)鍵字指定分隔符。

7.1數(shù)組的存儲

savetxt

用法：savetxt(“文件名.txt”，數(shù)組名，fmt=數(shù)組格式，delimiter=';')

補充：數(shù)組格式可以為整數(shù)型（%d）浮點型（%f）

from numpy import *
m1 = array([arange(1,4),arange(4,7),arange(7,10),arange(10,13),arange(13,16),arange(16,19)]) # 創(chuàng)建6行3列的二維數(shù)組
print(m1)
print("-"*50) # 分割線
savetxt("文件名.txt",m1,fmt='%d',delimiter=';')

運行結(jié)果1

運行結(jié)果2（分隔符為空格）

7.2數(shù)組的讀取

loadtxt

用法：新數(shù)組 = loadtxt(“文件名.txt”，dtype=‘數(shù)組格式'，delimiter=',')

補充：數(shù)組格式可以為整數(shù)型(int）浮點型（float）

from numpy import *
print("-"*50) # 分割線
a = loadtxt("文件名.txt",dtype=int,delimiter=';')
print(a) # 將數(shù)組轉(zhuǎn)置
print("-"*50) # 分割線

運行結(jié)果

總結(jié)

savetxt在保存數(shù)據(jù)時也可以將普通數(shù)組中的分隔符改為空格

利用這個方法我們可以將帶有逗號的二維數(shù)組改為分隔符為空格或者分號的二維數(shù)組。

例：用列表的方法寫一個二維數(shù)組

# 二維數(shù)組創(chuàng)建
from numpy import *
num = [[0 for j in range(1,5)]for i in range(1,4)]
print(num) # 打印數(shù)組
print("-"*50) # 分割線
num[2][1]=3 # 將第三行第二個元素修改為3
print(num) # 打印數(shù)組
print("-"*50) # 分割線
savetxt("a.txt",num,fmt='%d',delimiter=' ') # 將數(shù)組保存為分隔符為空格的形式
b = loadtxt("a.txt",dtype=int,delimiter=' ') # 將數(shù)組讀取出來賦給新數(shù)組
print(b) # 打印新數(shù)組
print("-"*50) # 分割線

運行結(jié)果