将当前目录下的所有图片名保存在csv中

import os
import csv
import pandas as pd
#要读取的文件的根目录
root_path=r'F:\emotionet\Img'
img = os.listdir( root_path )
print(len(img))
train=pd.DataFrame(data=img)
train.to_csv('F:\emotionet/train1.csv',index=0,header=0)#index是行标,header是列标,有中文的话就加encoding='gbk'
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
CSV 文件的处理步骤和文本文件类似,只需要在读取文件时指定分隔符和编码方式即可。下面是一个简单的示例代码,假设 CSV 文件有一列为 `comment`,保存了用户的评论信息: ```python import csv import jieba from wordcloud import WordCloud import matplotlib.pyplot as plt # 读取 CSV 文件并进行分词 comments = [] with open('data.csv', newline='', encoding='utf-8') as csvfile: reader = csv.DictReader(csvfile) for row in reader: comments.append(row['comment']) words = jieba.cut(' '.join(comments)) # 生成词云 wordcloud = WordCloud(font_path='msyh.ttc', background_color='white').generate(' '.join(words)) # 保存图片 plt.imshow(wordcloud, interpolation='bilinear') plt.axis('off') plt.savefig('wordcloud.png') ``` 在上面的代码,我们首先使用 `csv.DictReader` 函数读取 CSV 文件,并提取出评论信息。然后,我们使用 `jieba.cut` 函数对所有评论进行分词,并使用 `WordCloud` 类生成词云图。最后,我们使用 `pyplot.savefig` 函数将词云保存图片。 如果你想生成图片词云,可以使用以下代码: ```python import csv import jieba from wordcloud import WordCloud, ImageColorGenerator import matplotlib.pyplot as plt from PIL import Image # 读取 CSV 文件并进行分词 comments = [] with open('data.csv', newline='', encoding='utf-8') as csvfile: reader = csv.DictReader(csvfile) for row in reader: comments.append(row['comment']) words = jieba.cut(' '.join(comments)) # 生成词云 mask = np.array(Image.open('mask.png')) wordcloud = WordCloud(font_path='msyh.ttc', background_color='white', mask=mask).generate(' '.join(words)) image_colors = ImageColorGenerator(mask) # 保存图片 plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation='bilinear') plt.axis('off') plt.savefig('image_wordcloud.png') ``` 在这个代码,我们使用了与前面类似的方法生成图片词云。不同之处在于,我们使用 `csv.DictReader` 函数读取 CSV 文件,并提取出评论信息。 注意,这里需要将字体文件 `msyh.ttc` 和图片文件 `mask.png` 放在当前目录下。同时,你需要将 `data.csv` 替换为你自己的 CSV 文件
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值