遍历输出当前路径下的npy文件

##把信息输出到控制台

import numpy as np
import os

path = os.getcwd()
dirs = os.listdir(path)
for i in dirs:
    if os.path.splitext(i)[1] == ".npy":
        pre_train = np.load(i, allow_pickle=True, encoding="latin1")

        print("------name--------")
        print(os.path.splitext(i)[0])
        print("------shape-------")
        print(pre_train.shape)
        print("------data--------")
        print(pre_train)

##把信息保存到txt文件

path = os.getcwd()
dirs = os.listdir(path)
f = open('1.txt','a')
for i in dirs:
    if os.path.splitext(i)[1] == ".npy":
        pre_train = np.load(i, allow_pickle=True, encoding="latin1")

        print("------name--------", file = f)
        print(os.path.splitext(i)[0], file = f)
        print("------shape-------", file = f)
        print(pre_train.shape, file = f)
        print("------data--------", file = f)
        print(pre_train, file = f)
        print('\n', file = f)
f.close()