使用NumPy和pandas对CSV文件进行写操作的实例

yipeiwu_com5年前Python基础

数组存储成CSV之类的区隔型文件:

下面代码给随机数生成器指定种子,并生成一个3*4的NumPy数组

将一个数组元素的值设为NaN:

 In [26]: import numpy as np
 In [27]: np.random.seed(42)
 In [28]: a = np.random.randn(3,4)
 In [29]: a[2][2] = np.nan
 In [30]: print(a)
 [[ 0.49671415 -0.1382643  0.64768854 1.52302986]
 [-0.23415337 -0.23413696 1.57921282 0.76743473]
 [-0.46947439 0.54256004     nan -0.46572975]]

NumPy的savetxt()函数是与loadtxt()相对应的一个函数,它能以诸如CSV之类的区隔型文件格式保存数组:

In [31]: np.savetxt('np.csv',a,fmt='%.2f',delimiter=',',header="#1,#2,#3,#4")

上面的函数调用中,我们规定了用以保存数组的文件的名称、数组、可选格式、间隔符和一个可选的标题

通过cat np.csv,可以查看刚才所建的np.csv文件的具体内容

利用随机数组来创建pandas DataFrame:

 In [38]: df = pd.DataFrame(a)
 In [39]: df
 Out[39]: 
  0     1     2     3
 0 0.496714 -0.138264 0.647689 1.523030
 1 -0.234153 -0.234137 1.579213 0.767435
 2 -0.469474 0.542560    NaN -0.465730

pandas会自动替我们给数据取好列名

利用pandas的to_csv()方法可以为CSV文件生成一个DataFrame:

In [40]: df.to_csv('pd.csv',float_format='%.2f',na_rep="NAN!")

对于这个方法,我们需要提供文件名、类似于NumPy的savetxt()函数的格式化参数的可选格式串和一个表示NaN的可选字符串

以上这篇使用NumPy和pandas对CSV文件进行写操作的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

Python应用03 使用PyQT制作视频播放器实例

Python应用03 使用PyQT制作视频播放器实例

最近研究了Python的两个GUI包,Tkinter和PyQT。这两个GUI包的底层分别是Tcl/Tk和QT。相比之下,我觉得PyQT使用起来更加方便,功能也相对丰富。这一篇用PyQT实...

Appium Python自动化测试之环境搭建的步骤

Appium Python自动化测试之环境搭建的步骤

Appium简介 Appium是一个自动化测试开源工具,支持IOS和Android平台上的移动原生应用、移动Web应用和混合应用。所谓的“移动原生应用”是指那些用IOS或者Android...

Python利用matplotlib生成图片背景及图例透明的效果

Python利用matplotlib生成图片背景及图例透明的效果

前言 最近工作中遇到一个需求,在使用matplotlib生成图片,想要背景透明,而且图例部分也显示透明效果,通过查找相关资料找到了大概的设置方法,特此记录,方便自己或者有需要的朋友们参考...

OpenCV3.0+Python3.6实现特定颜色的物体追踪

OpenCV3.0+Python3.6实现特定颜色的物体追踪

一、环境 win10、Python3.6、OpenCV3.x;编译器:pycharm5.0.3 二、实现目标 根据需要追踪的物体颜色,设定阈值,在视频中框选出需要追踪的物体。 三、实现步...

Python中如何获取类属性的列表

前言 最近工作中遇到个需求是要得到一个类的静态属性,也就是说有个类 Type ,我要动态获取 Type.FTE 这个属性的值。 最简单的方案有两个: getattr(Type, 'F...