numpy常用方法

numpy中文文档和帮助

菜鸟教程numpy
numpy中文文档

---

实验中的函数

实验中经常读写CSV文件，我将读写的方法分装成一个函数，进行调用。

• 读取指定csv文件

def read_file(name):
    """ 读取csv文件
    
    参数说明：
        name : 文件名称
    """
    df = pd.read_csv(name)
    data = np.array(df)
    return data

• 写入csv文件

def write_file(name, data):
    """ 写入csv文件
    
    参数说明：
        name : 文件名称
        data : 写入数据
    """
    df = pd.DataFrame(data)
    df.to_csv(name, index=False)

• 一列数据进行归一化

def norm_weight(data):  
    """ 权重值归一化
    
    参数说明：
        data : 数据格式为一行numpy数组
        axis=0，对每一列进行操作。
        axis=1，对每一行进行操作。
    """
    sumdata = np.sum(data, axis=1)  # 对权重值求和
    result = []  # 记录归一化结果
    for i in range(len(data[0])):
        result.append(data[0][i] / sumdata[0])
    return result

一些使用到的方法

• 数组对应元素相乘

"""
参数说明：
	a : ndarray 数组
	b : ndarray 数组
"""
	c = a * b

• 数组相乘

"""
参数说明：
	a : ndarray 数组
	b : ndarray 数组
	out : ndarray, 可选，用来保存dot()的计算结果
"""
	numpy.dot(a, b, out=None) 

• 矩阵拼接

"""
	np.vstack:按垂直方向（行顺序）堆叠数组构成一个新的数组
	a和b的数组格式和大小必须一致
"""

	c = np.vstack((a,b))


"""
	np.hstack:按水平方向（列顺序）堆叠数组构成一个新的数组
	a和b的数组格式和大小必须一致
"""
	c = np.hstack((a,b))

• 次方和开根

"""
参数说明：
	a : ndarray 数组
	number : 为你要计算次方的数，2就为开平方，1/2就为开根
"""
	c  = np.power(a , number)

数据属性

• 维度

	print(np.shape(a))

• 总个数

	print(np.size(a))

• 元素类型

	print(np.dtype(a))