Hello Junqing, and welcome to the community!
Thank you so much! That is an old code, so I have to look around a bit to find it.
In the meanwhile, if you are in need of a quick solution, sheeny has even more complete figures already available that you can find here: My current Python Report for code examples