系统：Windows 7
语言版本：Anaconda3-4.3.0.1-Windows-x86_64
编辑器：pycharm-community-2016.3.2
- 这个系列讲讲Python的科学计算版块
- 今天讲讲pandas模块：获取DataFrame部分行
Part 1：示例
- 已知一个
DataFrame，想获取其中满足条件的行 - 从结果中可以知道，只保留了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：部分代码解读
-
df_2 = df[df["quality_1"].isin(list_1)]，从代码中可以看出，是以quality_1列作为筛选条件的，取quality_1列值为["pos_1", "pos_2", "pos_3"] -
df = pd.DataFrame(dict_1, columns=["quality_1", "measure_value", "up_tol", "down_tol"])由字典创建DataFrame，并且指定了列的排序
本文为原创作品，欢迎分享朋友圈
长按图片识别二维码，关注本公众号
Python 优雅 帅气
12x0.8.jpg
