python 循环创建变量

数据分析时，我们经常会有这样的需求：将数据集按某一个维度拆分成若干数据子集、创建多个变量。如果手动去实现，工作量较大。我们可以通过locals()配合循环语句来实现这一需求。我们以鸢尾花数据集为例，我们按照花的种类将数据集拆分成数据子集，分别放到不同的变量中去。实现代码如下：

from sklearn.datasets import load_iris

iris = load_iris()
X = iris.data
y = iris.target

for i in range(3):
    locals()['iris_'+str(i)] = X[y == i]

参考文献：

https://cloud.tencent.com/developer/ask/53504

https://www.cnblogs.com/wanxsb/archive/2013/05/07/3064783.html