Python使用numpy进行数据转换

一.测试数据

　　

二.代码实现

 # -*- coding: utf-8 -*-
 """
 Created on Sun Jul  7 11:35:01 2019
 加载数据时对指定列进行转换
 @author: zhen
 """

 import numpy as np
 np.set_printoptions(suppress=True) # 禁止numpy按科学计数法展示数据

 def color2num(str):
     dict = {"居民用电":220, "工业用电":380, "高压":11000, "超高压":500000, "特高压":1000000}
     return dict[str]

 data = np.loadtxt("data.csv", delimiter=",", converters={1:color2num}, encoding="gbk")
 print(data)

三.结果