简介
~~有一个神奇的事情,我看过很多人用a[a<3]
这样的语法来筛选数据,但是每次我用的时候都会报错~
直到今天,我仔细调试了一下才发现,该语句是有适用范围的。
比如说:矩阵可以用,但列表不可以。
下面做几个小的演示。
测试程序
l = [1,2,3,4,5] #测试列表
l[l<3] # 语句不会通过,会报错
l_n = np.array(l)
l_n[l_n<3] # 输出:[1, 2]
总结
- 矩阵适用该语句
l_n[l_n<3]
- 列表不适用
若要寻找列表中满足条件的数,可以用循环:
l = [1,2,3,4,5] #测试列表
[i for i in l if i < 3] # 输出:[1,2]