python numpy和list查询其中某个数的个数及定位方法
版权申诉
127 浏览量
2020-12-25
06:59:41
上传
评论 1
收藏 33KB PDF 举报
python numpy和和list查询其中某个数的个数及定位方法查询其中某个数的个数及定位方法
1. list
查询个数：查询个数：
调用list.count(obj)函数，返回obj在list中的个数。
输入：输入：
list_a = [2 for x in range(5)] print(list_a)
a_count = list_a.count(2)
print(a_count)
输出：输出：
[2, 2, 2, 2, 2]
定位元素：定位元素：
调用list.index(obj)函数，返回待查找对象第一个匹配项的位置。
输入：输入：
#!/usr/bin/python
aList = [123, 'xyz', 'zara', 'abc'];
print "Index for xyz : ", aList.index( 'xyz' ) ;
print "Index for zara : ", aList.index( 'zara' ) ;
输出：输出：
Index for xyz : 1
Index for zara : 2
2. numpy
查询个数：查询个数：
调用numpy的sum函数：np.sum( ndarray == 1 )
输入：输入：
import numpy as np
a = np.ones((4, 5))
print(a)
print(np.sum(a == 1))
输出：输出：
[[ 1. 1. 1. 1. 1.] [ 1. 1. 1. 1. 1.] [ 1. 1. 1. 1. 1.] [ 1. 1. 1. 1. 1.]] 20
定位元素：定位元素：
调用numpy的where函数：np.where( ndarray == obj )
输入：输入：
a = np.ones((4, 5))
a[1:3,1:3] = 4
print(a)
print(np.sum(a == 1))
print(np.where(a == 4))
输出：输出：
weixin_38625442
- 粉丝: 6
- 资源: 950
评论0