data_analysis18

笔记

import numpy as np

us_data = "./data/US_video_data_numbers.csv"
uk_data = "./data/GB_video_data_numbers.csv"

# 加载国家数据
us_data = np.loadtxt(us_data, delimiter=",", dtype=int)
uk_data = np.loadtxt(uk_data, delimiter=",", dtype=int)

# 添加国家信息
# 构造全为0的数据
zeros_data = np.zeros((us_data.shape[0], 1)).astype(int)
ones_data = np.ones((uk_data.shape[0], 1)).astype(int)

# 分别添加一列全为0,1的数组
us_data = np.hstack((us_data, zeros_data))
uk_data = np.hstack((uk_data, ones_data))

# 拼接两组数据
final_data = np.vstack((us_data, uk_data))
print(final_data)
print("*" * 66)

n1 = np.zeros((3, 4))
n2 = np.ones((3, 4))
n3 = np.eye(6)
print(n1)
print(n2)
print(n3)
print("*" * 66)

t = np.eye(4)
print(t)
t_max = np.argmax(t, axis=0)
print(t_max)

t[t == 1] = -1

print(t)
t_min = np.argmin(t, axis=1)
print(t_min)