在 pandas 中,可以使用以下几种方法来获取列:
1. 通过列名获取列:可使用类似于字典的方式通过列名获取列。例如,df[‘column_name’] 或 df.column_name。
import pandas as pd data = {'A': [1, 2, 3], 'B': [4, 5, 6]} df = pd.DataFrame(data) # 通过列名获取列 column_A = df['A'] print(column_A) # 或者 column_B = df.B print(column_B)
2. 使用 iloc 或 loc 获取列:可以使用 iloc 和 loc 方法获取列,其中 iloc 是基于索引位置的方式,而 loc 是基于标签的方式。
import pandas as pd data = {'A': [1, 2, 3], 'B': [4, 5, 6]} df = pd.DataFrame(data) # 使用 iloc 获取列 column_A = df.iloc[:, 0] # 第 0 列 print(column_A) # 使用 loc 获取列 column_B = df.loc[:, 'B'] # B 列 print(column_B)
3. 使用 .iloc[] 或 .loc[] 获取多列:还可以通过使用 .iloc[] 或 .loc[] 获取多列,传入一个列表包含所需列的索引位置或标签。
import pandas as pd data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]} df = pd.DataFrame(data) # 使用 iloc 获取多列 columns_A_B = df.iloc[:, [0, 1]] # 第 0 和第 1 列 print(columns_A_B) # 使用 loc 获取多列 columns_B_C = df.loc[:, ['B', 'C']] # B 和 C 列 print(columns_B_C)
通过以上方法,可以轻松地获取 pandas DataFrame 中的列数据。
版权申明:财旺号所有作品(图文、音视频)均由用户自行上传分享,仅供网友学习交流,不声明或保证其内容的正确性,如发现本站有涉嫌抄袭侵权/违法违规的内容。请发送邮件至 1790309299@qq.com 举报,一经查实,本站将立刻删除。