今天小编就为大家分享一篇python实现在多维数组中挑选符合条件的全部元素，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

问题产生：今天在编写神经网络的Cluster作业时，需要根据根据数据标签用不同的颜色画出数据的分布情况，由此学习到了这种高效的方法。 传统思路：用for循环来挑选符合条件的元素，这样十分浪费时间。 代码示例： from sklearn.datasets.samples_generator import make_blobs import numpy as np import matplotlib.pyplot as plt #product 20 samples and divide them