擅长:python、mysql、java
<p>我将用<code>os.listdir()</code>列出目录,并用<code>pandas</code>创建CSV:</p>
<pre><code>import os
import pandas as pd
os.chdir(r'cats_and_dogs_dataset')
folders = ['dogs', 'cats']
files = []
for folder in folders:
for file in os.listdir(folder):
files.append([file, folder])
pd.DataFrame(files, columns=['files', 'target']).to_csv('files_and_targets.csv')
</code></pre>
<p>CSV将如下所示:</p>
<pre><code> file target
0 dog.4001.jpg dogs
1 dog.4002.jpg dogs
2 dog.4003.jpg dogs
3 dog.4004.jpg dogs
4 dog.4005.jpg dogs
</code></pre>