python使用 pandas 读写 CSV 文件

python 经常使用 pandas 处理CSV文件的读写。

以下是读取CSV中列Label =2 且 x, y, z 列，得到 numpy 类型

import numpy as np
import pandas as pd
def parse_csv(csv_path):
    """ 读取CSV文件，输出成numpy类型的点集"""
    assert os.path.isfile(csv_path)
    csv_data = pd.read_csv(csv_path)
    center_line = np.array(csv_data[csv_data['label'] == 2][['x', 'y', 'z']])
    print(r'center_line points shape {}'.format(center_line.shape))
    return center_line

下面的例子则是将 numpy 保存成 CSV 文件。

import numpy as np
import pandas as pd
array_data = np.array([[1, 2, 3], [1, 2, 3]])
pd_data = pd.DataFrame(array_data)
pd_data.to_csv(r'D:\data.csv')