Python使用numpy进行数据转换

一.测试数据

　　

二.代码实现

 1 # -*- coding: utf-8 -*-
 2 """
 3 Created on Sun Jul  7 11:35:01 2019
 4 加载数据时对指定列进行转换
 5 @author: zhen
 6 """
 7 
 8 import numpy as np
 9 np.set_printoptions(suppress=True) # 禁止numpy按科学计数法展示数据
10 
11 def color2num(str):
12     dict = {"居民用电":220, "工业用电":380, "高压":11000, "超高压":500000, "特高压":1000000}
13     return dict[str]
14 
15 data = np.loadtxt("data.csv", delimiter=",", converters={1:color2num}, encoding="gbk")
16 print(data)

三.结果