Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import mysql.connector
- import numpy as np
- from colormath.color_objects import LabColor
- from colormath.color_diff import delta_e_cie2000
- import pandas as pd
- cnx = mysql.connector.connect(user='user', password='pass', host='localhost', database='database')
- cursor = cnx.cursor()
- def delta(cie_Lab1, cie_Lab2):
- cie_Lab1 = cie_Lab1[1:]
- cie_Lab1 = cie_Lab1[:-1]
- cie_Lab2 = cie_Lab2[1:]
- cie_Lab2 = cie_Lab2[:-1]
- CIE_list1 = cie_Lab1.split(",")
- CIE_list2 = cie_Lab2.split(",")
- #print CIE_list1
- CIE_L1 = CIE_list1[0]
- CIE_a1 = CIE_list1[1]
- CIE_b1 = CIE_list1[2]
- CIE_L2 = CIE_list2[0]
- CIE_a2 = CIE_list2[1]
- CIE_b2 = CIE_list2[2]
- color1 = LabColor(lab_l=CIE_L1, lab_a=CIE_a1, lab_b=CIE_b1)
- color2 = LabColor(lab_l=CIE_L2, lab_a=CIE_a2, lab_b=CIE_b2)
- deltae = delta_e_cie2000(color1, color2, Kl=1, Kc=1, Kh=1)
- return deltae
- selectstmt = 'SELECT CIE_Lab, ID FROM `database`.`table` LIMIT 1000'
- cursor.execute(selectstmt)
- color = cursor.fetchall()
- df = pd.DataFrame(columns = color, index = color)
- headers = df.dtypes.index
- sides = df.index
- print headers[1][0]
- print sides[0][0]
- nums = df.shape[0]
- for num in range(nums):
- for mun in range(nums):
- df.ix[mun][num] = delta(headers[num][0], sides[mun][0])
- print df
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement