python:numpy数组的拼接(完整代码)

import numpy as np

# csv数据路径
us_path = "./us.csv"
uk_path = "./us.csv"

# 用numpy载入数据，
us_data = np.loadtxt(us_path, delimiter=",", dtype=int)
uk_data = np.loadtxt(uk_path, delimiter=",", dtype=int)

# 添加1列全为0何全为1的数据，1代表uk，0代表us
us_zeros = np.zeros((us_data.shape[0],1)).astype(int)    # 构建一列全为0 的数组
uk_ones = np.ones((uk_data.shape[0],1)).astype(int)       # 构建一列全为1 的数组

#print(us_zeros)                  # 及时输出，看代码处理i结果是否为希望结果
#print("-" * 100)
#print(uk_ones)

# 水平拼接
us_data_new = np.hstack((us_zeros,us_data))   
uk_data_new = np.hstack((uk_ones,uk_data))

#print(us_data_new)
#print("*" * 100)
#print(uk_data_new)

# 垂直拼接
us_uk_vstack_data = np.vstack((us_data_new,uk_data_new))

print(us_uk_vstack_data)

输出的结果为：

[[        0  38999411  12597031        38       724]
 [        0   9667808  46983244        26      2059]
 [        0  36445855  35706526        46      2989]
 ...
 [        1  82531293 154917371         6      5996]
 [        1  72610963 182245183        12      8310]
 [        1 155305748 209305602       177      1648]]