python的pandas应用_Python小白的日常之Pandas简单应用

一：问题描述

今天在实验室遇到一个问题，需要将磁场模拟软件模拟的线圈磁场强度分布的数值导出来，之后用于绘制图像。

得到的数据是记录在一个txt文档里的。数据结构如下图所示：表头已经过简单的删除处理，并添加了熟悉的符号

现在需要将Y列为0的磁场强度H值导出，以绘制线圈中心线的磁场变化曲线图。

然而数据行数有4万多行，用常规方法显然不够迅速！

二：曲折的解决问题

首先。打开Python3.6，本小白用的是Anaconda3集成开发环境，已经安装了pandas库。输入以下代码导入pandas库：# -*- coding: utf-8 -*-

import pandas as pd

第二步，读取.txt格式的文件df = pd.read_table('0.24.txt', sep='\s+')

'0.24.txt'是目标文件名，注意将.py文件保存在相同路径下。

sep是读取时的分割方式，注意到txt文档不像EXCEL那样严格的分列和行，因此可以将空格作为分行分列的依据。

'\s+'是正则表达式，\s代表空格，+代表根据贪心匹配，也就是匹配一个或多个空格。

现在将dfprint一下，看看怎么样：print(df)

果然厉害，得到了以0~N的列表以及以txt文件第一行为行标的数据结构。(DataFrame数据结构)

接下来，需要从这些数据中找出我想要的。

例如想找出Y列的值为在-18之间的所有数据，可以用以下代码：df1 = df[df.Y == -18]

但若要找出Y列的值为在-18~-15之间的所有数据，下列代码是行不通的：df1 = df[df.Y>-18 and df.Y

我最初的想法是，来两次选择:df1 = df[df.Y > -18]

df2 = df1[df1.Y

print(df2)

这样分步来做是可行的。作为小白，达到了目的我就没去想太多了。

找到了想要的数据集之后，现在得做关键一步，就是把想要的H列数据输出出来，好复制到绘图软件进行绘制。PS：这里指定要用ORIGIN绘图，否则就用Matplotlib绘图了

在df2打印出来时，会提示你这堆数据是几行几列(200row*4column)，据此可以用一个循环导出H列数据：for i in range(0,200,1):

H = df1.iloc[i,3]

print(H)

注意iloc的下表是从0开始的。

据此，便达到目的了。

有更简单的方法，我会在下面补充增加！

我是小白，痛苦但很实用。