Python-科学计算-pandas-01-df获取部分数据

微信公众号原文

系统：Windows 7
语言版本：Anaconda3-4.3.0.1-Windows-x86_64
编辑器：pycharm-community-2016.3.2

• 这个系列讲讲Python的科学计算版块
• 今天讲讲pandas模块：获取DataFrame部分行

Part 1：示例

1. 已知一个DataFrame，想获取其中满足条件的行
2. 从结果中可以知道，只保留了df中的前3行数据

执行结果

1.png

Part 2：代码

import pandas as pd

dict_1 = {"quality_1": ["pos_1", "pos_2", "pos_3", "pos_4", "pos_5"],
          "measure_value": [6, 4, 6, 3.5, 2.5],
          "up_tol": [5, 5, 3, 3, 2],
          "down_tol": [-5, -5, -3, -3, 2]}

list_1 = ["pos_1", "pos_2", "pos_3"]

df = pd.DataFrame(dict_1, columns=["quality_1", "measure_value", "up_tol", "down_tol"])
print(df)

print("\n只取需要的数据：")
df_2 = df[df["quality_1"].isin(list_1)]
print(df_2)

代码截图

2.png

Part 3：部分代码解读

1. df_2 = df[df["quality_1"].isin(list_1)]，从代码中可以看出，是以quality_1列作为筛选条件的，取quality_1列值为["pos_1", "pos_2", "pos_3"]
2. df = pd.DataFrame(dict_1, columns=["quality_1", "measure_value", "up_tol", "down_tol"])由字典创建DataFrame，并且指定了列的排序

本文为原创作品，欢迎分享朋友圈

长按图片识别二维码，关注本公众号
Python 优雅 帅气

12x0.8.jpg