如何在python中查看维度名

在Python中，如果你有一个Pandas DataFrame或NumPy数组，你可以通过`.columns`或`.shape`属性查看其列名（对于DataFrame）或维度名（对于数组）。如果是一个xarray.DataArray，则可以使用`.dims`属性获取维度名。

例如，对于Pandas DataFrame:

import pandas as pd

df = pd.DataFrame({
    'column1': [1, 2, 3],
    'column2': [4, 5, 6]
})
print(df.columns)  # 输出：Index(['column1', 'column2'], dtype='object')

for dim in df.shape:  # 也可以这样查看维度
    print(dim)

对于xarray.DataArray:

import xarray as xr

data_array = xr.DataArray([1, 2, 3], dims=['time'])
print(data_array.dims)  # 输出：('time',)

如果你不确定数据的具体类型，可以用`type()`函数检查：

if isinstance(obj, pd.DataFrame):
    # 处理DataFrame
elif isinstance(obj, np.ndarray):
    # 处理NumPy数组
elif isinstance(obj, xr.DataArray):
    # 处理xarray.DataArray
else:
    print("未知的数据类型")