making_csv.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os#, os.path\n",
    "import natsort\n",
    "import csv\n",
    "#import glob\n",
    "import re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "file_path = 'test/'\n",
    "file_lists = os.listdir(file_path)\n",
    "file_lists = natsort.natsorted(file_lists)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "drop-1264.png\n",
      "drop-1265.png\n",
      "drop-1266.png\n",
      "drop-1267.png\n",
      "drop-1268.png\n",
      "drop-1269.png\n",
      "drop-1270.png\n",
      "drop-1271.png\n",
      "drop-1272.png\n",
      "drop-1273.png\n",
      "drop-1274.png\n",
      "drop-1275.png\n",
      "drop-1276.png\n",
      "drop-1277.png\n",
      "drop-1278.png\n",
      "drop-1279.png\n",
      "drop-1280.png\n",
      "drop-1281.png\n",
      "drop-1282.png\n",
      "drop-1283.png\n",
      "drop-1284.png\n",
      "drop-1285.png\n",
      "drop-1286.png\n",
      "drop-1287.png\n",
      "drop-1288.png\n",
      "drop-1289.png\n",
      "drop-1290.png\n",
      "drop-1291.png\n",
      "drop-1292.png\n",
      "drop-1293.png\n",
      "drop-1294.png\n",
      "drop-1295.png\n",
      "drop-1296.png\n",
      "drop-1297.png\n",
      "drop-1298.png\n",
      "drop-1299.png\n",
      "drop-1300.png\n",
      "drop-1301.png\n",
      "drop-1302.png\n",
      "drop-1303.png\n",
      "drop-1304.png\n",
      "drop-1305.png\n",
      "drop-1306.png\n",
      "drop-1307.png\n",
      "drop-1308.png\n",
      "drop-1309.png\n",
      "drop-1310.png\n",
      "drop-1311.png\n",
      "drop-1312.png\n",
      "drop-1313.png\n",
      "drop-1314.png\n",
      "drop-1315.png\n",
      "drop-1316.png\n",
      "drop-1317.png\n",
      "drop-1318.png\n",
      "drop-1319.png\n",
      "drop-1320.png\n",
      "drop-1321.png\n",
      "drop-1322.png\n",
      "drop-1323.png\n",
      "drop-1324.png\n",
      "drop-1325.png\n",
      "drop-1326.png\n",
      "drop-1327.png\n",
      "drop-1328.png\n",
      "drop-1329.png\n",
      "drop-1330.png\n",
      "drop-1331.png\n",
      "drop-1332.png\n",
      "drop-1333.png\n",
      "drop-1334.png\n",
      "drop-1335.png\n",
      "drop-1336.png\n",
      "drop-1337.png\n",
      "drop-1338.png\n",
      "drop-1339.png\n",
      "drop-1340.png\n",
      "drop-1341.png\n",
      "drop-1342.png\n",
      "drop-1343.png\n",
      "drop-1344.png\n",
      "drop-1345.png\n",
      "drop-1346.png\n",
      "drop-1347.png\n",
      "drop-1348.png\n",
      "drop-1349.png\n",
      "drop-1350.png\n",
      "drop-1351.png\n",
      "drop-1352.png\n",
      "drop-1353.png\n",
      "drop-1354.png\n",
      "drop-1355.png\n",
      "drop-1356.png\n",
      "drop-1357.png\n",
      "drop-1358.png\n",
      "drop-1359.png\n",
      "drop-1360.png\n",
      "drop-1361.png\n",
      "drop-1362.png\n",
      "drop-1363.png\n",
      "drop-1364.png\n",
      "drop-1365.png\n",
      "drop-1366.png\n",
      "drop-1367.png\n",
      "drop-1368.png\n",
      "drop-1369.png\n",
      "drop-1370.png\n",
      "drop-1371.png\n",
      "drop-1372.png\n",
      "drop-1373.png\n",
      "drop-1374.png\n",
      "drop-1375.png\n",
      "drop-1376.png\n",
      "drop-1377.png\n",
      "drop-1378.png\n",
      "drop-1379.png\n",
      "drop-1380.png\n",
      "drop-1381.png\n",
      "drop-1382.png\n",
      "drop-1383.png\n",
      "drop-1384.png\n",
      "drop-1385.png\n",
      "drop-1386.png\n",
      "drop-1387.png\n",
      "drop-1388.png\n",
      "drop-1389.png\n",
      "drop-1390.png\n",
      "drop-1391.png\n",
      "drop-1392.png\n",
      "drop-1393.png\n",
      "drop-1394.png\n",
      "drop-1395.png\n",
      "drop-1396.png\n",
      "drop-1397.png\n",
      "drop-1398.png\n",
      "drop-1399.png\n",
      "drop-1400.png\n",
      "drop-1401.png\n",
      "drop-1402.png\n",
      "drop-1403.png\n",
      "motor-1409.png\n",
      "motor-1410.png\n",
      "motor-1411.png\n",
      "motor-1412.png\n",
      "motor-1413.png\n",
      "motor-1414.png\n",
      "motor-1415.png\n",
      "motor-1416.png\n",
      "motor-1417.png\n",
      "motor-1418.png\n",
      "motor-1419.png\n",
      "motor-1420.png\n",
      "motor-1421.png\n",
      "motor-1422.png\n",
      "motor-1423.png\n",
      "motor-1424.png\n",
      "motor-1425.png\n",
      "motor-1426.png\n",
      "motor-1427.png\n",
      "motor-1428.png\n",
      "motor-1429.png\n",
      "motor-1430.png\n",
      "motor-1431.png\n",
      "motor-1432.png\n",
      "motor-1433.png\n",
      "motor-1434.png\n",
      "motor-1435.png\n",
      "motor-1436.png\n",
      "motor-1437.png\n",
      "motor-1438.png\n",
      "motor-1439.png\n",
      "motor-1440.png\n",
      "motor-1441.png\n",
      "motor-1442.png\n",
      "motor-1443.png\n",
      "motor-1444.png\n",
      "motor-1445.png\n",
      "motor-1446.png\n",
      "motor-1447.png\n",
      "motor-1448.png\n",
      "motor-1449.png\n",
      "motor-1450.png\n",
      "motor-1451.png\n",
      "motor-1452.png\n",
      "motor-1453.png\n",
      "motor-1454.png\n",
      "motor-1455.png\n",
      "motor-1456.png\n",
      "motor-1457.png\n",
      "motor-1458.png\n",
      "motor-1459.png\n",
      "motor-1460.png\n",
      "motor-1461.png\n",
      "motor-1462.png\n",
      "motor-1463.png\n",
      "motor-1464.png\n",
      "motor-1465.png\n",
      "motor-1466.png\n",
      "motor-1467.png\n",
      "motor-1468.png\n",
      "motor-1469.png\n",
      "motor-1470.png\n",
      "motor-1471.png\n",
      "motor-1472.png\n",
      "motor-1473.png\n",
      "motor-1474.png\n",
      "motor-1475.png\n",
      "motor-1476.png\n",
      "motor-1477.png\n",
      "motor-1478.png\n",
      "motor-1479.png\n",
      "motor-1480.png\n",
      "motor-1481.png\n",
      "motor-1482.png\n",
      "motor-1483.png\n",
      "motor-1484.png\n",
      "motor-1485.png\n",
      "motor-1486.png\n",
      "motor-1487.png\n",
      "motor-1488.png\n",
      "motor-1489.png\n",
      "motor-1490.png\n",
      "motor-1491.png\n",
      "motor-1492.png\n",
      "motor-1493.png\n",
      "motor-1494.png\n",
      "motor-1495.png\n",
      "motor-1496.png\n",
      "motor-1497.png\n",
      "motor-1498.png\n",
      "motor-1499.png\n",
      "motor-1500.png\n",
      "motor-1501.png\n",
      "motor-1502.png\n",
      "motor-1503.png\n",
      "motor-1504.png\n",
      "motor-1505.png\n",
      "motor-1506.png\n",
      "motor-1507.png\n",
      "motor-1508.png\n",
      "motor-1509.png\n",
      "motor-1510.png\n",
      "motor-1511.png\n",
      "motor-1512.png\n",
      "motor-1513.png\n",
      "motor-1514.png\n",
      "motor-1515.png\n",
      "motor-1516.png\n",
      "motor-1517.png\n",
      "motor-1518.png\n",
      "motor-1519.png\n",
      "motor-1520.png\n",
      "motor-1521.png\n",
      "motor-1522.png\n",
      "motor-1523.png\n",
      "motor-1524.png\n",
      "motor-1525.png\n",
      "motor-1526.png\n",
      "motor-1527.png\n",
      "motor-1528.png\n",
      "motor-1529.png\n",
      "motor-1530.png\n",
      "motor-1531.png\n",
      "motor-1532.png\n",
      "motor-1533.png\n",
      "motor-1534.png\n",
      "motor-1535.png\n",
      "motor-1536.png\n",
      "motor-1537.png\n",
      "motor-1538.png\n",
      "motor-1539.png\n",
      "motor-1540.png\n",
      "motor-1541.png\n",
      "motor-1542.png\n",
      "motor-1543.png\n",
      "motor-1544.png\n",
      "motor-1545.png\n",
      "motor-1546.png\n",
      "motor-1547.png\n",
      "motor-1548.png\n",
      "motor-1549.png\n",
      "motor-1550.png\n",
      "motor-1551.png\n",
      "motor-1552.png\n",
      "motor-1553.png\n",
      "motor-1554.png\n",
      "motor-1555.png\n",
      "motor-1556.png\n",
      "motor-1557.png\n",
      "motor-1558.png\n",
      "motor-1559.png\n",
      "motor-1560.png\n",
      "motor-1561.png\n",
      "motor-1562.png\n",
      "motor-1563.png\n",
      "motor-1564.png\n",
      "water-1123.png\n",
      "water-1124.png\n",
      "water-1125.png\n",
      "water-1126.png\n",
      "water-1127.png\n",
      "water-1128.png\n",
      "water-1129.png\n",
      "water-1130.png\n",
      "water-1131.png\n",
      "water-1132.png\n",
      "water-1133.png\n",
      "water-1134.png\n",
      "water-1135.png\n",
      "water-1136.png\n",
      "water-1137.png\n",
      "water-1138.png\n",
      "water-1139.png\n",
      "water-1140.png\n",
      "water-1141.png\n",
      "water-1142.png\n",
      "water-1143.png\n",
      "water-1144.png\n",
      "water-1145.png\n",
      "water-1146.png\n",
      "water-1147.png\n",
      "water-1148.png\n",
      "water-1149.png\n",
      "water-1150.png\n",
      "water-1151.png\n",
      "water-1152.png\n",
      "water-1153.png\n",
      "water-1154.png\n",
      "water-1155.png\n",
      "water-1156.png\n",
      "water-1157.png\n",
      "water-1158.png\n",
      "water-1159.png\n",
      "water-1160.png\n",
      "water-1161.png\n",
      "water-1162.png\n",
      "water-1163.png\n",
      "water-1164.png\n",
      "water-1165.png\n",
      "water-1166.png\n",
      "water-1167.png\n",
      "water-1168.png\n",
      "water-1169.png\n",
      "water-1170.png\n",
      "water-1171.png\n",
      "water-1172.png\n",
      "water-1173.png\n",
      "water-1174.png\n",
      "water-1175.png\n",
      "water-1176.png\n",
      "water-1177.png\n",
      "water-1178.png\n",
      "water-1179.png\n",
      "water-1180.png\n",
      "water-1181.png\n",
      "water-1182.png\n",
      "water-1183.png\n",
      "water-1184.png\n",
      "water-1185.png\n",
      "water-1186.png\n",
      "water-1187.png\n",
      "water-1188.png\n",
      "water-1189.png\n",
      "water-1190.png\n",
      "water-1191.png\n",
      "water-1192.png\n",
      "water-1193.png\n",
      "water-1194.png\n",
      "water-1195.png\n",
      "water-1196.png\n",
      "water-1197.png\n",
      "water-1198.png\n",
      "water-1199.png\n",
      "water-1200.png\n",
      "water-1201.png\n",
      "water-1202.png\n",
      "water-1203.png\n",
      "water-1204.png\n",
      "water-1205.png\n",
      "water-1206.png\n",
      "water-1207.png\n",
      "water-1208.png\n",
      "water-1209.png\n",
      "water-1210.png\n",
      "water-1211.png\n",
      "water-1212.png\n",
      "water-1213.png\n",
      "water-1214.png\n",
      "water-1215.png\n",
      "water-1216.png\n",
      "water-1217.png\n",
      "water-1218.png\n",
      "water-1219.png\n",
      "water-1220.png\n",
      "water-1221.png\n",
      "water-1222.png\n",
      "water-1223.png\n",
      "water-1224.png\n",
      "water-1225.png\n",
      "water-1226.png\n",
      "water-1227.png\n",
      "water-1228.png\n",
      "water-1229.png\n",
      "water-1230.png\n",
      "water-1231.png\n",
      "water-1232.png\n",
      "water-1233.png\n",
      "water-1234.png\n",
      "water-1235.png\n",
      "water-1236.png\n",
      "water-1237.png\n",
      "water-1238.png\n",
      "water-1239.png\n",
      "water-1240.png\n",
      "water-1241.png\n",
      "water-1242.png\n",
      "water-1243.png\n",
      "water-1244.png\n",
      "water-1245.png\n",
      "water-1246.png\n",
      "water-1247.png\n"
     ]
    }
   ],
   "source": [
    "f = open('test.csv', 'w', encoding='utf-8')\n",
    "wr = csv.writer(f)\n",
    "wr.writerow([\"Img_name\", \"Class\"])\n",
    "for file_name in file_lists:\n",
    "    print(file_name)\n",
    "    wr.writerow([file_name, re.sub('-\\d*[.]\\w{3}', '', file_name)])\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#dir_list = glob.glob(os.path.join('dataset', '*'))\n",
    "#print(dir_list)\n",
    "#f = open('soundee_label_2.csv', 'w', encoding='utf-8')\n",
    "#wr = csv.writer(f)\n",
    "#wr.writerow([\"Img_name\", \"Class\"])\n",
    "#for dir in dir_list:\n",
    "#    file_list = list(glob.glob(dir+'/*.png', recursive=True))\n",
    "#    count = len(file_list)\n",
    "#    dir = dir.replace(\"dataset/\", \"\")\n",
    "#    print(count)\n",
    "#    for file in file_list:\n",
    "#        wr.writerow([file, dir])\n",
    "#f.close()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}