cluster_test1.py 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. #go through csv file, speichern x und y
  2. #dann loop durch alle anderen x und y:
  3. # if abstand x weniger als
  4. # if abstand y weniger als
  5. # dann selber cluster
  6. # alles in ein file/variable speichern und werte löschen??? oder markieren als bereits geclustert
  7. # alles durchgehen
  8. #wenn alles durch dann neues x und y und nochmal alles von vorn
  9. import csv
  10. #csvfile = open('values.csv','r')
  11. #spamreader1 = list(csv.reader(csvfile, dialect='excel', delimiter=','))
  12. #csvFileArray = []
  13. csvfile1 = open('values.csv', 'r')
  14. spamreader1 = list(csv.reader(csvfile1, dialect='excel', delimiter=','))
  15. csvfile1.close()
  16. #print(spamreader1)
  17. #csvfile = open('values.csv', 'r+')
  18. new_rows_list = []
  19. with open("values.csv", "r") as csvfile:
  20. reader = csv.reader(csvfile, delimiter=',')
  21. for row in reader:
  22. #print(row)
  23. x = row[5]
  24. #print(x)
  25. y = row[6]
  26. #print(y)
  27. for row1 in spamreader1:
  28. #print(row1)
  29. x1 = row1[5]
  30. #print(x1)
  31. y1 = row1[6]
  32. #print(abs(float(x1) - float(x)))
  33. #print(abs(float(y1) - float(y)))
  34. if (abs(float(x1) - float(x)) < 21.0) and (abs(float(y1) - float(y)) < 40.0): # and row[7] == False:
  35. #print(row)
  36. row[4] = row[4] + " " + row1[4]
  37. print(row[4])
  38. row[7] = True
  39. writer = csv.writer(csvfile)
  40. writer.writerow(row)
  41. new_row = [row[0], row[1], row[2],row[3],row[4],row[5],row[6]] #write all values, including new merged text
  42. new_rows_list.append(new_row)
  43. csvfile.close()
  44. file2 = open("merged_values.csv", 'wb')
  45. writer = csv.writer(file2)
  46. writer.writerows(new_rows_list)
  47. file2.close()