diff --git a/src/xtgeo/well/_welldata.py b/src/xtgeo/well/_welldata.py index 2aa41aaa2..634fa9207 100644 --- a/src/xtgeo/well/_welldata.py +++ b/src/xtgeo/well/_welldata.py @@ -272,3 +272,21 @@ def get_dataframe_copy(self, infer_dtype: bool = False): else: return self.data.copy() + + def get_dataframe(self): + """Get the dataframe, . + + If infer_dtype is True, then DISC columns will be of "int32" type + """ + + if infer_dtype: + dfr = self.data.copy() + + for name, wtype in self.wlogtypes.items(): + if "DISC" in wtype: + dfr[name] = dfr[name].astype("int32") + + return dfr + + else: + return self.data.copy()