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')