题解 | 多项式特征的Phi变换

import numpy as np

def phi_transform(data: list[float], degree: int) -> list[list[float]]:
	if degree < 0 or not data:
		return []
	else:
         m = []
         for i in data:
             for j in range(degree + 1):
                 m.append(i ** j)
         n = np.array(m, dtype=float)
         return n.reshape(-1, degree + 1).tolist()



if __name__ == "__main__":
    data = eval(input())
    degree = int(input())
    print(phi_transform(data, degree))