正文

python中怎么选取两列

小六 /47 浏览量
1224

在Python中，获取数据表或矩阵中的特定列可以通过多种方式实现，以下是使用pandas库和NumPy库提取两列数据的方法：

使用pandas库

 import pandas as pd 创建一个包含两列数据的字典 data = { 'col1': [1, 2, 3, 4, 5], 'col2': ['a', 'b', 'c', 'd', 'e'] } 将字典转换为DataFrame df = pd.DataFrame（data） 提取两列到一个新的DataFrame new_df = df[['col1', 'col2']] print（new_df）

使用NumPy库

 import numpy as np 创建一个NumPy数组 sz = np.array（[[12, 54, 64, 54, 84], [54, 84, 21, 35, 64], [45, 235, 48, 564, 485]]）  使用列表推导式提取第二列数据 lis = [row for row in sz] print（lis）

以上代码分别展示了如何使用pandas和NumPy库从数据结构中提取特定的列。如果您需要从文本中提取数据，可以使用正则表达式，例如：

 import re text = "1,2,3\n4,5,6\n7,8,9" 使用正则表达式提取每行的第一列和第三列 pattern = re.compile（r'（\d）,（\d）,（\d）'） 对文本进行匹配，并提取数字 matches = pattern.findall（text） 将提取的数字转换为整数列表 numbers = [list（map（int, match）） for match in matches] print（numbers）

请根据您的具体需求选择合适的方法

-- 展开阅读全文 --