Numpy 笔记

最近在使用Numpy的时候经常查阅一些文档，看一些函数的使用方法或参数，可能比较入门，但是依然需要做一些笔记以备日后再次查阅。

1.where()的几种用法

有以下两个小问题：

（1）有如下的两个数组a和b，如何获取a和b相同元素的位置？即输出 array([1, 3, 5, 8]

a = np.array([0,1,2,3,4,5,6,7,8,9])

b = np.array([1,1,1,3,1,5,1,1,8,1])

（2）有如下的数组a，如何获取5到10之间的所有元素？即输出 (array([6, 9, 10]),)

a = np.array([1, 6, 2, 7, 10, 3, 45])

想到了使用 where()解决这个问题，但是 where()具体都可以传递什么参数呢？请看以下：
1）只给 where()传递一个数组作为参数，返回其中非0元素的下标。

2）给where()传递一个包含True/False值的数组，返回该数组中True值的下标，结合numpy数组的关系运算，可以返回数组中符合特定条件的元素的下标。

3）给where()传递一个条件数组和两个值或数组，对于条件数组中等价于True的位置，从第一个值或数组中取值进行替换，否则从第二个值或数组中取值进行替换。

想到了另一个问题：如何在不改变a数组的前提下将a数组中的所有奇数替换为 -1？

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

out = np.where(arr % 2 == 1, -1, a)

2.集合运算

依然是先抛出一个问题：如何获取两个numpy数组之间的公共项？
numpy中几个函数专门针对集合进行运算。

import numpy as np

# 返回排序的、去重的两个数组的并集
np.union1d(x, y)
# 返回排序的、去重的两个数组的交集
np.intersect1d(x, y)
# 返回在x中不在y中的元素，即差集  
np.setdiff1d(x, y)
# 返回x和y的异或集
np.setxor1d(x, y)

3.打印问题