numpy学习

numpy学习

---

---

前言

---

一、numpy对二维数组的切片操作

1.获取数组所有元素

lst = [[1,2,3,4,5],
       [6,7,8,9,10]]
lst = np.array(lst)
# 使用切片操作
result = lst[:,:]

2.去掉最后一列

# 去掉最后一列
result = lst[:,:-1]

2.获取最后一列，或者第几列和第几列

# 获取最后一列
result = lst[:,[-1]]
# 获取最后两列
result = lst[:,[-2,-1]]

2.对字符串获取文件名操作

# 获取json文件名
dir_path = "E:/recovery_source_code/Movement_Classification/Datasets/a.json"
result = dir_path[len("E:/recovery_source_code/Movement_Classification/Datasets/"):]
# 修改文件名后缀
kpts_file_name = result.split('.')[0]+".txt" 
print(kpts_file_name)

pkl文件如何保存并加载

import pickle

'''
可以将模型参数保存为pkl文件，在不重新训练的情况下使用加载模型参数进行预测或进一步的分析是。
'''

# 创建一个字典对象
data = {
    'name': 'Alice',
    'age': 30,
    'city': 'New York'
}


# 打开一个文件用于写入
with open('data.pkl', 'wb') as f:
    # 使用pickle.dump()将字典对象序列化并保存到文件中
    pickle.dump(data, f)
# 打开一个文件用于读取
with open('data.pkl', 'rb') as f:
    # 使用pickle.load()从文件中读取序列化的对象并还原为原来的Python对象
    loaded_data = pickle.load(f)

# 打印加载的数据
print(loaded_data)

总结

持续添加内容进行更新学习。