https://teerinfo.com/how-to-get-column-names-from-a-dataframe-in-python/