# -*- coding = utf-8 -*-
# @Time : 2022/1/6 19:47
# @Author : 军
# @File : test2.py
# @Software : PyCharm

import numpy as np

a=[1,2,3,4]
b=np.array(a)
print(b)

array_zero=np.zeros([10,10])
array_one=np.ones([10,10])
print(array_zero)
print(array_one)

arr=np.array([[1,2,3],[4,5,6]])
print(arr)
print(1/arr)
print(arr*arr)
print(arr**0.5)

arr=np.arange(10)
print(arr)
print(arr[5])
print(arr[5:8])
arr[5:8]=12
print(arr)
arr_slice=arr[5:8]
arr_slice[1]=12345
print(arr)
arr_slice[:]=64
print(arr)#改切片会影响到原数组

arr3d=np.array([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]])
print(arr3d)
print(arr3d[0][1])

arr=np.random.randn(5,4)#正态分布数据
print(arr)
print(arr.mean())
print(arr.sum())

print(arr.mean(axis=1))
print(arr.sum(0))

arr=np.array([[0,1,2],[3,4,5],[6,7,8]])
print(arr.cumsum(0))
print(arr.cumprod(1))

x=np.array([[1,2,3],[4,5,6]])
y=np.array([[6.,23.],[-1,7],[8,9]])
print(x.dot(y))


def numpy_sum(n):
    a=np.array(n)**2
    b=np.array(n)**3
    return a+b

print(numpy_sum(10))