当前位置： 移动技术网 > IT编程>脚本编程>Python > Python Numpy-基础教程

Python Numpy-基础教程

2019年01月04日  | 移动技术网IT编程  | 我要评论

乐山社区,橙领学堂,女子花90万买二手房

目录

1. 为什么要学习numpy?

  • numpy可以对整个array进行复杂计算，而不需要像list一样写loop
  • 它的ndarray提供了快速的基于array的数值运算
  • memory-efficient container that provides fast numerical operations
  • 学习pandas的必备

证明numpy比list优秀:

import numpy as np
my_arr = np.arange(1000000)
my_list = list(range(1000000))

%time for _ in range(10): my_arr2 = my_arr * 2                  # wall time: 25 ms
%time for _ in range(10): my_list2 = [x * 2 for x in my_list]   # wall time: 933 ms

2. numpy基本用法

2.1. 创建np.ndarry

注意： numpy只能装同类型的数据

# method 1: np.array()
## 1-d
a = np.array([1,2,3])
a.shape
a.dtype  # int32, boolean, string, float
a.ndim

## 2-d
a = np.array([[0,1,2],[3,4,5]])

# method 2:使用函数(arange, linspace, ones, zeros, eys, diag,random)创建  
a = np.arange(10)
a = np.linspace(0,1,6, endpoint=false)
a = np.ones((3,3))
a = np.zeros((3,3))
a = np.eye(3)  
a = np.diag(np.array([1,2,3,4]))  
a = np.triu(np.ones((3,3)),1)

# method 3: random values
a = np.random.rand(4)   # unifomr in [0,1]
a = np.random.randn(4) # gaussian
np.random.seed(1234)

2.2. indexing and slicing

  • slice create a view on the original array(change will affect original array)
# 1-d
a = np.arange(10)
a[5], a[-1]    # index: 4,9 
a[5:8] = 12    # slice: all 5-8 is set as 12  
arr[5:8].copy()   # slice without view  

# 2-d
a = np.ones((3,3))
a[2]           # second row
a[2].copy()    # slice without view
a[0][2]        # special value  

a[:2]
a[:2, 1:] = 0

boolean index

names = np.array(['bob', 'joe', 'will', 'bob', 'will', 'joe', 'joe'])
data = np.random.randn(7, 4)

data[names == 'bob']  # select a row from data based on the if names equals bob(boolean value)
data[~(names == 'bob')]    # not equal to bob
data[(names == 'bob') | (names == 'will')]    #e qual to bob and will
data[data<0] = 0

2.3. universal functions

a function that performs element-wise operations on data in ndarrays

a = np.arange(10)  
b = np.arange(2,12)

# single
a + 1
a*2 
np.sqrt(a)
np.exp(a)
np.sin(a)

# binary
a>b                     # return boolean ndarray
np.array_equal(a,b)     # eual? 
np.maximum(a, b)        # find max value between each pair values
np.logical_or(a,b)      # attentions, a and b must be boolean array

2.4. array-oriented

  • probelm 1

we wished to evaluate the function `sqrt(x^2 + y^2)`` across a regular grid of values.

the np.meshgrid function takes two 1d arrays and produces two 2d matrices corresponding to all pairs of (x, y) in the two arrays:

points = np.arange(-5, 5, 0.01) # 1000 equally spaced points
xs, ys = np.meshgrid(points, points)
z = np.sqrt(xs ** 2 + ys ** 2)

import matplotlib.pyplot as plt
%matplotlib inline

plt.imshow(z, cmap=plt.cm.gray); plt.colorbar()
plt.title("image plot of $\sqrt{x^2 + y^2}$ for a grid of values")

  • problem 2

we have two array(x,y) and one boolean array, we want select x if boolean=true, while select y if boolean=false->np.where()

xarr = np.array([1.1, 1.2, 1.3, 1.4, 1.5])
yarr = np.array([2.1, 2.2, 2.3, 2.4, 2.5])
cond = np.array([true, false, true, true, false])
result = np.where(cond, xarr, yarr)        # array([1.1, 2.2, 1.3, 1.4, 2.5])

np.where的后面两个参数可以是array，数字. 是数字的话就可以做替换工作，比如我们将随机生成的array中大于0的替换为2，小于0的替换为-2

arr = np.random.randn(4, 4)
np.where(arr > 0, 2, -2)     # 大于0改为2，小于0改为-2
np.where(arr > 0, 2, arr)    # 大于0改为2，小于0不变

2.5. mathematical operations

a = np.random.randn(5, 4)  
np.mean(a)
np.mean(a, axis = 1)
np.sum(a)
a.consum()
a.sort()
a.argmax()  # index of maxium

names = np.array(['bob', 'joe', 'will', 'bob', 'will', 'joe', 'joe'])
np.unique(names)
sorted(set(names))


您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网