Numpy 学习 -- argwhere函数,random函数

argwhere

np.argwhere(expression)

返回满足表达式的元素下标

random

np.random.choice(a, size=None, replace=True, p=None)

从给定的一维数组a生成size个随机样本，按照概率p抽取

举例：

 np.random.choice(5, 3)
    array([0, 3, 4])

利用pandas和numpy将样本分成训练集和测试集

sample = np.random.choice(processed_data.index, size=int(len(processed_data)*0.9), replace=False)
train_data, test_data = processed_data.iloc[sample], processed_data.drop(sample)

np.random.shuffle(array)

对原有的array做一个随机排序

np.random.randn((dim0,dim1,…,dimn))

返回一个服从正态分布的序列，序列的size为（dim0,dim1,…,dimn）