feat: added code for splitting nemeth matrice

This commit is contained in:
Krzysztof Rudnicki 2024-11-28 18:08:06 +01:00
parent 0eea9ba5a2
commit 8a9c4fc2c8
2 changed files with 19 additions and 0 deletions

2
.gitattributes vendored
View File

@ -2,3 +2,5 @@ code/poli3.npz filter=lfs diff=lfs merge=lfs -text
code/nemeth12.npz filter=lfs diff=lfs merge=lfs -text
code/spd_5000.npz filter=lfs diff=lfs merge=lfs -text
code/spd_10000.npz filter=lfs diff=lfs merge=lfs -text
code/nemeth12_1.npz filter=lfs diff=lfs merge=lfs -text
code/nemeth12_2.npz filter=lfs diff=lfs merge=lfs -text

17
split_nemeth.py Normal file
View File

@ -0,0 +1,17 @@
import numpy as np
# Load the .npz file
input_file = 'code/nemeth12.npz'
data = np.load(input_file)
# Create dictionaries to hold the split data
data_keys = list(data.keys())
split_point = len(data_keys) // 2
first_half = {key: data[key] for key in data_keys[:split_point]}
second_half = {key: data[key] for key in data_keys[split_point:]}
# Save the halves to separate .npz files
np.savez('nemeth12_1.npz', **first_half)
np.savez('nemeth12_2.npz', **second_half)
print(f"Data split into 'nemeth12_1.npz' and 'nemeth12_2.npz'")