欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

使用numpy来创建数组

程序员文章站 2022-05-28 19:38:22
...

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、更多创建数组的函数

使用numpy来创建数组

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

相关标签: numpy