Python Numpy Tutorials: 数组--4
程序员文章站
2022-10-10 21:12:18
# -*- coding: utf-8 -*-
"""
python version: 3.5
created on thu may 11...
# -*- coding: utf-8 -*- """ python version: 3.5 created on thu may 11 14:42:49 2017 e-mail: eric2014_lv@sjtu.edu.cn @author: didilv """ import numpy as np a = np.array([[1,2], [3, 4], [5, 6]]) # 寻找元素大于2的值,并且返回其bool值 bool_idx = (a > 2) # find the elements of a that are bigger than 2; # this returns a numpy array of booleans of the same # shape as a, where each slot of bool_idx tells # whether that element of a is > 2. print(bool_idx) # prints "[[false false] # [ true true] # [ true true]]" # we use boolean array indexing to construct a rank 1 array # consisting of the elements of a corresponding to the true values # of bool_idx print(a[bool_idx]) # prints "[3 4 5 6]" # we can do all of the above in a single concise statement: print(a[a > 2]+10) # prints "[13 14 15 16]"