用python实现:一个文件夹下有10个csv,对于每个csv,第一列是点云数量,第二列是类别,统计所有csv中点云数量大于0的,不同类别的目标个数有多少
时间: 2023-12-23 08:02:51 浏览: 30
可以使用Python的pandas库结合os模块来读取文件夹下的多个csv文件,并进行数据处理。以下是代码示例:
```python
import os
import pandas as pd
# 定义结果字典
result = {}
# 遍历文件夹下的csv文件
for filename in os.listdir('foldername'):
if filename.endswith('.csv'):
# 读取csv文件
df = pd.read_csv(os.path.join('foldername', filename))
# 筛选点云数量大于0的数据
df = df[df['点云数量'] > 0]
# 统计不同类别的目标个数
count = df.groupby('类别')['点云数量'].count()
# 将结果合并到字典中
for key in count.keys():
if key not in result:
result[key] = 0
result[key] += count[key]
# 输出结果
print(result)
```
其中,`foldername`是待读取的文件夹名,`df = pd.read_csv(os.path.join('foldername', filename))`是用来读取文件夹下的多个csv文件,`df[df['点云数量'] > 0]`语句是用来筛选出点云数量大于0的数据,`df.groupby('类别')['点云数量'].count()`是用来对不同类别的目标进行计数。最后,将结果合并到字典中,并输出统计结果即可。