Python - ¿Cómo puedo crear un archivo CSV desde python?

Lo que pasa es que tengo que resolver una matriz A que tenga una dimensión n = 10-100 valores y una columna B que los valores van hacer igual a la dimensión de la matriz que genere.

Mi duda es que para realizar esta actividad tengo que generar dos archivos CSV uno para la matriz A y otro para la columna B, pero en esa parte me genera problemas, alguien que me pueda orientar en como generarla. Saludos

haber si te sirve esta manera: https://www.lawebdelprogramador.com/foros/Python/1609359-Crear-un-archivo-csv-a-partir-de-una-lista.html#i1609443