遍历数组的固定部分

Iterating Over fixed sections of Arrays

下午好, 我可以就如何处理我拥有的 Python 代码部分获得建议吗?我有一个 1000 列宽的数组,我希望将 .h5 文件应用于每个部分。例如:

data_len=list(range(100,1100,300)) 
score_set = np.zeros((len(data_len),2))
ind = 0
for end_pt in data_len:

# ITERATE OVER .H5 FILES
directory = r'/media/Thesis_Maps/test_runs_2/h5_files'

for file in os.listdir(directory): #setup for loop to read each .h5 file
    filename=os.fsdecode(file)
    if filename.endswith('.h5'):
        fpath=os.path.join(directory,filename)
        model = load_model(fpath)
        #model.summary()
        X_run = X.iloc[:,0:end_pt]
        y_run = y.iloc[:,0:end_pt] 

我希望将 1000 行分成 4 组,每组 100、300、300 和 300。但是,我认为这不是这样做,而是创建 100、400、700 和 1000 的组,这是不正确的。问题是,如何将 1000 列宽的行分解为 100、300、300 和 300 个数据点,并为每个数据点应用一个唯一的 .h5 文件?任何建议都会很棒,谢谢。

您可以修正您的限制:

import numpy as np

data_len=list(range(100,1100,300)) 
score_set = np.zeros((len(data_len),2))
ind = 0
so_far = 0
for end_pt in data_len:
    end_pt -= so_far   # fix your list above
    so_far += end_pt   # remember how much so far
    print(end_pt)

输出:

100
300
300
300