iis服务器助手广告广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Python如何用NumPy读取和保存点云数据
  • 439
分享到

Python如何用NumPy读取和保存点云数据

2024-04-02 19:04:59 439人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

目录前言loadtxt函数的用法基本用法指定每一列的数据类型结合生成器使用tofile和fromfile函数前言 最近在学习点云处理的时候用到了Modelnet40数据集,该数据集总

前言

最近在学习点云处理的时候用到了Modelnet40数据集,该数据集总共有40个类别,每个样本的点云数据存放在一个TXT文件中,每行的前3个数据代表一个点的xyz坐标。我需要把TXT文件中的每个点读取出来,然后用Open3D进行显示。怎么把数据从TXT文件中读取出来呢?NumPy提供了一个功能非常强大的函数loadtxt可以非常简单地实现这个功能。来看一下代码:

import open3d as o3d
import numpy as np

def main():
    points_data = np.loadtxt("airplane_0001.txt", delimiter=",", dtype=np.float32)
    pcd = o3d.geometry.PointCloud()
    pcd.points = o3d.utility.Vector3dVector(points_data[:, :3])
    o3d.visualization.draw_geometries([pcd])

if __name__ == '__main__':
    main()

从上面的代码可以看到,只需要一行代码就可以把TXT文件中的点云数据读取进来了,接下来就可以调用Open3D的接口进行显示了。在介绍loadtxt函数的用法之前,

顺便看一下Open3D的显示效果:

loadtxt函数的用法

基本用法

在上面的例子中,由于TXT里面每一行的数据是用逗号分割的,所以在调用loadtxt函数的时候除了设置文件路径外,还需要设置参数delimiter=","。另外,该函数默认的数据类型为float64,如果是其他数据类型的话还需要设置dtype为对应类型。

points_data = np.loadtxt("airplane_0001.txt", delimiter=",") #没有指定数据类型
print('shape: ', points_data.shape)
print('data type: ', points_data.dtype)

结果:

shape:  (10000, 6)
data type:  float64 

指定每一列的数据类型

假如我们有一个CSV文件:

x,y,z,label,id
-0.098790,-0.182300,0.163800,1,1
0.994600,0.074420,0.010250,0.2,2
0.189900,-0.292200,-0.926300,3,3
-0.989200,0.074610,-0.012350,4,4

该文件前面3列的数据类型是浮点型,后面2列的数据类型为整型,那么按照前面的方式设置dtype来读取就不合适了。不过没关系,loadtxt函数可以设置每一列数据的数据类型,只不过稍微复杂一点,来看一下代码:

data = np.loadtxt("test.txt", delimiter=",",
                      dtype={'names': ('x', 'y', 'z', 'label', 'id'), 
                            'fORMats': ('f4', 'f4', 'f4', 'i4', 'i4')},
                      skiprows=1)
print('data: ', data)
print('data type: ', data.dtype)

这段代码的重点是dtype={}里面的内容,'names'用来设置每一列数据的名称,'formats'则用来设置每一列数据的数据类型,其中'f4'表示float32'i4'表示int32。另外,CSV文件中的第一行不是数据内容,可以设置参数skiprows=1跳过第一行的内容。

输出结果:

data:  [(-0.09879, -0.1823 ,  0.1638 , 1, 1) ( 0.9946 ,  0.07442,  0.01025, 0, 2)
 ( 0.1899 , -0.2922 , -0.9263 , 3, 3) (-0.9892 ,  0.07461, -0.01235, 4, 4)]
data type:  [('x', '<f4'), ('y', '<f4'), ('z', '<f4'), ('label', '<i4'), ('id', '<i4')]

可以看到,通过这样的方式设置dtype,读取的每一行数据变成了一个tuple类型。

结合生成器使用

NumPy的文档中可以知道,loadtxt函数的第一个参数可以是文件对象、文件名或者生成器。传入生成器有什么用呢?我们来看几个例子。

处理多个分隔符

假如我们的文件内容是这样的,每一行数据有3个分隔符",","/"和"-":

9.87,1.82,1.63,1/11-1
9.94,7.44,1.02,1/11-2
1.89,2.92,9.26,1/11-3
0.98,7.46,1.23,1/11-4

这种情况下不能通过delimiter参数设置多个分隔符,这时候就可以通过生成器来进行处理:

def generate_lines(file_path, delimiters=[]):
    with open("test.txt") as f:
        for line in f:
            line = line.strip()
            for d in delimiters:
                line = line.replace(d, " ")
            yield line

delimiters = [",", "/", "-"]
generator = generate_lines("test.txt", delimiters)
data = np.loadtxt(generator)
print(data)

这段代码构建了一个生成器将文件中每一行的分隔符全部替换成loadtxt函数默认的空格分隔符,然后把生成器传入loadtxt函数,这样loadtxt函数就能成功解析文件中的数据了。

输出结果:

[[ 9.87  1.82  1.63  1.   11.    1.  ]
 [ 9.94  7.44  1.02  1.   11.    2.  ]
 [ 1.89  2.92  9.26  1.   11.    3.  ]
 [ 0.98  7.46  1.23  1.   11.    4.  ]]

读取指定的行

在某些情况下,我们需要读取指定几行的数据,那么也可以通过生成器来实现。还是上面的文件内容,我们通过生成器来读取第2行和第3行:

def generate_lines(file_path, delimiters=[], rows=[]):
    with open("test.txt") as f:
        for i, line in enumerate(f):
            line = line.strip()
            for d in delimiters:
                line = line.replace(d, " ")
            if i in rows:
                yield line

delimiters = [",", "/", "-"]
rows = [1, 2]
generator = generate_lines("test.txt", delimiters, rows)
data = np.loadtxt(generator)
print(data)

输出结果:

[[ 9.94  7.44  1.02  1.   11.    2.  ]
 [ 1.89  2.92  9.26  1.   11.    3.  ]]

通过上面的例子可以知道,loadtxt函数结合生成器使用可以实现很多的功能。

tofile和fromfile函数

TXT文件中读取到点云数据后,我想把数据保存到二进制文件中,需要怎么操作呢?NumPyndarray类提供了tofile函数可以非常方便地将数据保存到二进制文件中。把数据以二进制文件保存后又怎么读进来呢?NumPy还提供了一个fromfile函数用于从文本文件和二进制文件中读取数据。

import open3d as o3d
import numpy as np

def main():
    points_data = np.loadtxt(
        "airplane_0001.txt", delimiter=",", dtype=np.float32)

    bin_file = 'airplane_0001.bin'
    points_data = points_data[:, :3]
    points_data.tofile(bin_file)

    pc = np.fromfile(bin_file, dtype=np.float32)
    pc = pc.reshape(-1, 3)
    pcd = o3d.geometry.PointCloud()
    pcd.points = o3d.utility.Vector3dVector(pc)
    o3d.visualization.draw_geometries([pcd])

if __name__ == '__main__':
    main()

在上面这段示例代码中,我从airplane_0001.txt文件中读取了点云数据,然后通过tofile函数将数据保存到二进制文件airplane_0001.bin中,再用fromfile函数从二进制文件中把点云数据读取出来用Open3D进行显示。

为了前后呼应,让我们换个角度再看一眼显示效果:

到此这篇关于python如何用NumPy读取和保存点云数据的文章就介绍到这了,更多相关Python NumPy 内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

--结束END--

本文标题: Python如何用NumPy读取和保存点云数据

本文链接: https://www.lsjlt.com/news/120231.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
  • Python如何用NumPy读取和保存点云数据
    目录前言loadtxt函数的用法基本用法指定每一列的数据类型结合生成器使用tofile和fromfile函数前言 最近在学习点云处理的时候用到了Modelnet40数据集,该数据集总...
    99+
    2024-04-02
  • pythonNumPy读取和保存点云数据实现
    目录前言loadtxt函数的用法基本用法指定每一列的数据类型结合生成器使用tofile和fromfile函数前言 最近在学习点云处理的时候用到了Modelnet40数据集,该数据集总...
    99+
    2024-04-02
  • python数据如何保存为npy和npz格式并读取
    这篇“python数据如何保存为npy和npz格式并读取”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“python数据如何保...
    99+
    2023-07-02
  • python怎么读取保存串口数据
    在Python中,可以使用pySerial库来读取和保存串口数据。以下是一个简单的示例代码: import serial # 打开...
    99+
    2024-04-02
  • Java如何使用jxl读取excel并保存到数据库
    小编给大家分享一下Java如何使用jxl读取excel并保存到数据库,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!项目中涉及到读取excel中的数据,保存到数据库...
    99+
    2023-05-31
    java jxl excel
  • python怎么存储和读取数据
    在Python中,可以使用各种方法来存储和读取数据,具体方式取决于数据的类型和使用环境。以下是一些常见的方法: 使用变量:可以将数...
    99+
    2023-10-26
    python
  • 如何在Python读取与存储数据
    这篇文章将为大家详细讲解有关如何在Python读取与存储数据,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。一、图示二、csv文件读取csv文件read_csv(file_path or bu...
    99+
    2023-06-15
  • Python如何存储和读取ASCII码形式的byte数据
    目录存储和读取ASCII码形式的byte数据Python ASCII码的获取存储和读取ASCII码形式的byte数据 Python可以存byte数据到txt,但不要用str的方式直接...
    99+
    2024-04-02
  • Python如何读取和写入Excel数据
    这篇文章主要介绍“Python如何读取和写入Excel数据”,在日常操作中,相信很多人在Python如何读取和写入Excel数据问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”Python如何读取和写入Exce...
    99+
    2023-06-02
  • Python 读取数据并保存为txt文件的方式
     首先是读取文件,这里使用了pandas库 import pandas as pddata_train = pd.read_csv("train_dataset.csv", encoding='utf-8',sep='|')train_co...
    99+
    2023-09-27
    python 开发语言
  • python中如何读取数据
    python 中读取数据的方法有:从文件读取(打开、读入、逐行读取)从文本流读取(创建、写入、重置指针、读取)从 csv 文件读取(创建读取器、逐行读取)从 json 文件读取(加载数据...
    99+
    2024-04-02
  • python 数据保存为npy和npz格式并读取的完整代码
    目录python 数据保存为npy格式python 数据保存为npz格式python 数据保存为npy格式 补充:numpy.save("./文件名", 数组名)...
    99+
    2024-04-02
  • python如何读取mongodb数据
    要在Python中读取MongoDB数据,您可以使用PyMongo库。以下是一个简单示例:首先,确保您已经安装了PyMongo库。您...
    99+
    2023-08-23
    python mongodb
  • Pandas常用的读取和保存数据的函数使用(csv,mysql,json,excel)
    pandas 是基于NumPy 的一种工具,该工具是为解决数据分析任务而创建的。Pandas 纳入了大量库和一些标准的数据模型,提供了高效地操作大型数据集所需的工具。Pandas的名...
    99+
    2024-04-02
  • Python爬虫如何获取数据并保存到数据库中
    本篇内容主要讲解“Python爬虫如何获取数据并保存到数据库中”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Python爬虫如何获取数据并保存到数据库中”吧!1.简介介绍-网络爬虫(又称为网页蜘...
    99+
    2023-07-02
  • Python点云处理(一)点云数据读取与写入
    目录 0 简述1 LAS/LAZ格式1.1 las/laz数据读取1.2 las/laz数据写入 2 PCD格式2.1 pcd格式读取2.2 pcd格式写入 3 PLY格式3.1 pl...
    99+
    2023-09-27
    python 3d 算法 数据可视化
  • 如何使用ASP索引来提高numpy存储的数据读取速度?
    numpy是Python中最常用的科学计算库之一。它提供了一个方便的接口来处理大量的数字数据,并且有着高效的计算能力。然而,当我们需要处理大规模的数据集时,可能会遇到一些性能问题。在本文中,我们将介绍如何使用ASP索引来提高numpy存储...
    99+
    2023-08-20
    索引 numpy 存储
  • Python如何使用StringIO和BytesIO读写内存数据
    本文小编为大家详细介绍“Python如何使用StringIO和BytesIO读写内存数据”,内容详细,步骤清晰,细节处理妥当,希望这篇“Python如何使用StringIO和BytesIO读写内存数据”文章能帮助大家解决疑惑,下面跟着小编的...
    99+
    2023-06-30
  • python如何读取串口数据
    在Python中,可以使用第三方库pyserial来读取串口数据。首先需要安装pyserial库,可以使用pip安装: pip in...
    99+
    2024-04-02
  • Spark中的数据读取保存和累加器实例详解
    目录数据读取与保存Text文件Sequence文件Object对象文件累加器累加器概念系统累加器数据读取与保存 Text文件 对于 Text文件的读取和保存 ,其语法和实现是最简单的...
    99+
    2022-11-13
    Spark数据读取保存累加器 Spark数据读取保存
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作