Array Processing – Python Numpy – How to work with Arrays in Python
Hi Guys,
Thanks for all your emails. In this note, we will be looking at Array Processing in Python.
This is part of lectures on Learning Python for Data Analysis and Visualization by Jose Portilla on Udemy.
In [3]:
|
1 2 3 |
import numpy as np import matplotlib.pyplot as plt %matplotlib inline |
In [4]:
|
1 |
points = np.arange(-5,5,0.01) |
In [5]:
|
1 |
dx, dy = np.meshgrid(points,points) |
In [6]:
|
1 |
dx |
Out[6]:
In [7]:
|
1 |
dy |
Out[7]:
In [8]:
|
1 2 |
z = (np.sin(dx) + np.sin(dy)) z |
Out[8]:
In [9]:
|
1 |
plt.imshow(z) |
Out[9]:
In [10]:
|
1 2 3 |
plt.imshow(z) plt.colorbar() plt.title('Plot of sin(x) + sin(y)') |
Out[10]:
In [11]:
|
1 2 3 |
#numpy where A = np.array([1,2,3,4]) B = np.array([100,200,300,400]) |
In [12]:
|
1 |
condition = np.array([True, True, False, False]) |
In [13]:
|
1 |
answer = [(A_val if cond else B_val) for A_val, B_val, cond in zip(A, B, condition)] |
In [14]:
|
1 |
answer |
Out[14]:
In [15]:
|
1 2 |
answer2 = np.where(condition,A,B) answer2 |
Out[15]:
In [16]:
|
1 |
from numpy.random import randn |
In [19]:
|
1 2 |
arr = randn(5,5) arr |
Out[19]:
In [20]:
|
1 |
np.where(arr<0,0,arr) |
Out[20]:
In [22]:
|
1 2 |
arr2 = np.array([[1,2,3],[4,5,6],[7,8,9]]) arr2 |
Out[22]:
In [23]:
|
1 |
arr2.sum() |
Out[23]:
In [24]:
|
1 |
arr2.sum(0) |
Out[24]:
In [29]:
|
1 |
arr2.sum(1) |
Out[29]:
Want more information like this?
