文章作者：Tyan
博客：noahsnail.com ?|? CSDN ?|? 簡書
本文主要是關(guān)于numpy的一些基本運算的用法。
#!/usr/bin/env python
# _*_ coding: utf-8 _*_
import numpy as np
# Test 1
a = np.arange(2, 14).reshape(3, 4)
print a
# Test 1 result
[[ 2 3 4 5]
[ 6 7 8 9]
[10 11 12 13]]
# Test 2
# 計算矩陣最小值的索引
print np.argmin(a)
# 計算矩陣最大值的索引
print np.argmax(a)
# 計算矩陣的均值
print np.mean(a)
print a.mean()
print np.average(a)
# 計算矩陣的中位數(shù)
print np.median(a)
# 計算前n項之和
print np.cumsum(a)
# 計算相鄰兩位的差
print np.diff(a)
# 找出非零的數(shù), 輸出的是非零數(shù)的索引，分別為行的索引和列的索引
print np.nonzero(a)
# Test 2 result
11
0
8.5
8.5
8.5
8.5
[ 14 27 39 50 60 69 77 84 90 95 99 102]
[[-1 -1 -1]
[-1 -1 -1]
[-1 -1 -1]]
(array([0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]), array([0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3]))
# Test 3
a = np.arange(14, 2, -1).reshape(3, 4)
print a
# 矩陣排序，按行排序
print np.sort(a)
# 矩陣的轉(zhuǎn)置
print np.transpose(a)
print a.transpose()
print a.T
print a.T.dot(a)
# Test 3 result
[[14 13 12 11]
[10 9 8 7]
[ 6 5 4 3]]
[[11 12 13 14]
[ 7 8 9 10]
[ 3 4 5 6]]
[[14 10 6]
[13 9 5]
[12 8 4]
[11 7 3]]
[[14 10 6]
[13 9 5]
[12 8 4]
[11 7 3]]
[[14 10 6]
[13 9 5]
[12 8 4]
[11 7 3]]
[[332 302 272 242]
[302 275 248 221]
[272 248 224 200]
[242 221 200 179]]
# Test 4
print a
# 矩陣的處理，所有小于5的數(shù)等于5，所有大于10的數(shù)等于10
print np.clip(a, 5, 10)
# 計算矩陣指定維度的均值, 0是列, 1是行
print np.mean(a, axis = 0)
# Test 4 result
[[14 13 12 11]
[10 9 8 7]
[ 6 5 4 3]]
[[10 10 10 10]
[10 9 8 7]
[ 6 5 5 5]]
[ 10. 9. 8. 7.]
