[snap] improve sinks
This commit is contained in:
+3
-16
@@ -1350,15 +1350,6 @@ class SnapshotProcessor(HDF5Container):
|
||||
return alpha_g
|
||||
|
||||
def _sinks(self):
|
||||
csv_name = (
|
||||
self.path
|
||||
+ "/output_"
|
||||
+ str(self.num).zfill(5)
|
||||
+ "/sink_"
|
||||
+ str(self.num).zfill(5)
|
||||
+ ".csv"
|
||||
)
|
||||
sinks = np.loadtxt(csv_name, delimiter=",")
|
||||
header = [
|
||||
"Id",
|
||||
"M",
|
||||
@@ -1379,14 +1370,10 @@ class SnapshotProcessor(HDF5Container):
|
||||
"int_lum",
|
||||
"Teff",
|
||||
]
|
||||
if len(sinks) == 0:
|
||||
sinks = np.zeros(len(header))
|
||||
csv_name = f"{self.path}/output_{self.num:05}/sink_{self.num:05}.csv"
|
||||
df = pd.read_csv(csv_name, header=None, names=header)
|
||||
return {key: df[key].values for key in df}
|
||||
|
||||
sinks_dict = {}
|
||||
for key, a in zip(header, sinks):
|
||||
sinks_dict[key] = a
|
||||
|
||||
return sinks_dict
|
||||
|
||||
def _pspec(self, **kwargs):
|
||||
outfile = self.pspec_filename
|
||||
|
||||
Reference in New Issue
Block a user