使用numpy来创建数组

1、通过python列表来创建数组

import numpy as np

if __name__ == "__main__":
    a = [1,2,3]
    arr1 = np.array(a)
    print(arr1)
    # [1 2 3]
    
    
    b = [[1,2,3],[4,5,6]]
    arr2 = np.array(b)
    print(arr2)
    '''
    [[1 2 3]
     [4 5 6]]
    '''

2、使用numpy的内置函数来创建数组

a、zeros函数创建全为0的数组

    #创建一个一行5列全为0的数组
    arr5 = np.zeros(5)
    print(arr5)
    #[ 0.  0.  0.  0.  0.]
    #创建一个两行三列全为0的数组
    arr6 = np.zeros((2,3))
    print(arr6)
    '''
    [[ 0.  0.  0.]
     [ 0.  0.  0.]] 
    '''

b、使用arange创建一个指定范围的数组

    #创建一个从2到6的数组
    '''
    start:数组开始的数字
    stop:数组结束的数字(不包括)
    step:步长
    '''
    arr7 = np.arange(start=2,stop=7,step=1)
    print(arr7)
    #[2 3 4 5 6]

c、使用empty创建数组

    arr8 = np.empty((2,3))
    print(arr8)
    '''
    [[  2.44032786e-312   3.56037961e-307   4.00536722e-307]
    [  2.33645827e-307   4.22787460e-307   3.56038640e-307]]
    '''

注意：使用empty函数创建出来的数组是没有经过初始化的，都是垃圾值。

d、更多创建数组的函数

3、查看numpy数组的信息

a、查看数组的行

    print(arr1.ndim)
    #1
    print(arr2.ndim)
    #2

b、查看数组的行列

    c = [[1],[2],[3]]
    arr3 = np.array(c)
    #第一个参数表示行，第二个参数表示列
    #注意一行三列和三行一列的区别
    print(arr1.shape)
    #(3,),一行三列
    print(arr3.shape)
    #(3, 1),三行一列
    print(arr2.shape)
    #(2, 3)

c、查看数组的数据类型

    #整数默认的是int32,浮点数默认的是float64
    print(arr1.dtype)
    #int32
    d = [1.,2.,3.]
    arr4 = np.array(d)
    print(arr4.dtype)
    #float64