import numpy as np

# Datos proporcionados
iron_butterfly_1 = [1730.00, 1730.00, 1732.50, 1722.50, 1702.50, 1717.50, 1717.50, 1715.00, 1715.00, 1715.00, 
                    1700.00, 1700.00, 1695.00, 1695.00, 1695.00, 1675.00, 1712.50, 1710.00, 1710.00, 1710.00, 
                    1710.00, 1670.00, 1705.00, 1705.00, 1705.00, 1695.00, 1695.00, 1715.00, 1715.00, 1720.00, 
                    1720.00, 1720.00]
iron_butterfly_2 = [1655.00, 1655.00, 1660.00, 1650.00, 1650.00, 1645.00, 1645.00, 1637.50, 1622.50, 1622.50, 
                    1632.50, 1627.50, 1630.00, 1620.00, 1620.00, 1630.00, 1627.50, 1632.50, 1632.50, 1607.50, 
                    1607.50, 1607.50, 1610.00, 1615.00, 1612.50, 1612.50, 1627.50, 1627.50, 1620.00, 1620.00, 
                    1620.00, 1630.00]
spx_price =        [5093.18, 5093.18, 5092.46, 5091.80, 5091.13, 5091.04, 5091.05, 5092.39, 5091.32, 5090.38,
                    5090.43, 5090.23, 5090.74, 5090.95, 5090.41, 5090.80, 5090.50, 5089.81, 5089.89, 5091.00, 
                    5091.07, 5090.65, 5090.34, 5091.31, 5091.24, 5091.59, 5091.54, 5091.45, 5091.63, 5090.71, 
                    5090.64, 5090.31]

# Convertir las listas en matrices numpy
iron_butterfly_1 = np.array(iron_butterfly_1)
iron_butterfly_2 = np.array(iron_butterfly_2)
spx_price = np.array(spx_price)

# Crear una matriz con los datos
data = np.array([iron_butterfly_1, iron_butterfly_2, spx_price])

# Calcular la matriz de correlación
correlation_matrix = np.corrcoef(data)

# La correlación entre el Iron Butterfly 1 y el SPX está en la posición (0, 2) de la matriz de correlación
correlation_iron_butterfly_1_spx = correlation_matrix[0, 2]

# La correlación entre el Iron Butterfly 2 y el SPX está en la posición (1, 2) de la matriz de correlación
correlation_iron_butterfly_2_spx = correlation_matrix[1, 2]

# La correlación entre el Iron Butterfly 1 y el Iron Butterfly 2 está en la posición (0, 1) de la matriz de correlación
correlation_iron_butterfly_1_2 = correlation_matrix[0, 1]

print("Correlación entre Iron Butterfly 1 y SPX:", correlation_iron_butterfly_1_spx)
print("Correlación entre Iron Butterfly 2 y SPX:", correlation_iron_butterfly_2_spx)
print("Correlación entre Iron Butterfly 1 y Iron Butterfly 2:", correlation_iron_butterfly_1_2)