Include curve unit in LASFile.df() column names

[issimo007]

Hello, I tried looking around all day but couldn't find a way to use the "las.df()" command in a way that would generate a dataframe header that includes the "mnemonic (unit)". If anyone has a suggestion it is greatly appreciated. Thank you!

[kinverarity1]

Thanks @issimo007 - that's a great suggestion for a feature! It's not available at the moment.

For now you could achieve what you want with this code instead of las.df()

def las_to_df_with_units(las):
    column_mapper = {
        c.mnemonic: f"{c.mnemonic} ({c.unit})"
        for c in las.curves
    }
    return (
        las.df()
        .reset_index()
        .rename(columns=column_mapper)
        .set_index(list(column_mapper.values())[0])
    )

import lasio.examples
las = lasio.examples.open("sample.las")
df = las_to_df_with_units(las)
print(df)

          DT (US/M)  RHOB (K/M3)  NPHI (V/V)  SFLU (OHMM)  SFLA (OHMM)  ILM (OHMM)  ILD (OHMM)
DEPT (M)                                                                                      
1670.000     123.45       2550.0        0.45       123.45       123.45       110.2       105.6
1669.875     123.45       2550.0        0.45       123.45       123.45       110.2       105.6
1669.750     123.45       2550.0        0.45       123.45       123.45       110.2       105.6

[issimo007]

Thank you Kent! I couldn't figure out how to access the units once imported into lasio. All the best, Mauro.

[kinverarity1]

This could be added to the LASFile.df method as a keyword argument e.g. LASFile.df(include_units=False).