Cómo guardar el VALOR del color en una tabla de atributos en QGIS para una capa vectorial con un renderizador de símbolos graduado

Question

Intento conseguir esto -> ¿Cómo copiar el estilo de color en una columna de color de la tabla de atributos de una capa?

guión de @Germán Carrillo

prefix = "'"
layer = iface.activeLayer()
attr = layer.rendererV2().classAttribute()
attrColor = 'color' # Name of the field to store colors
fieldIndex = layer.dataProvider().fieldNameIndex(attrColor)
attrFeatMap = {}

for cat in layer.rendererV2().categories(): 
  expr = "\""+attr+"\"="+prefix+unicode(cat.value())+prefix
  for f in layer.getFeatures(QgsFeatureRequest(QgsExpression(expr))):
    attrMap = { fieldIndex : cat.symbol().color().name()}
    attrFeatMap[ f.id() ] = attrMap

layer.dataProvider().changeAttributeValues( attrFeatMap )

pero después de usar ese script estoy recibiendo un error

Traceback (most recent call last):
  File "<input>", line 1, in <module>
AttributeError: 'QgsGraduatedSymbolRendererV2' object has no attribute 'categories'

porque es para una capa clasificada con un Renderizador de Símbolos Categorizados.

Entonces el problema es que puedo usar este script solo con color aleatorio... Pero, ¿cómo hacer lo mismo gradiente de colores como ndvi -1 / 1

¿podría esto ayudarme? -> http://docs.qgis.org/testing/en/docs/pyqgis_developer_cookbook/vector.html#graduated-symbol-renderer

Primera vez que trabajo con python.....

EDIT : Qgis 2.10.1 - PISA EDIT : Añadir imagen

Answer 1

El problema es que la clase de renderizador es QgsCategorizedSymbolRendererV2 en lugar de QgsGraduatedSymbolRendererV2 . Así que categoricé el siguiente shapefile por el campo FID:

y se ejecutó un código idéntico (el script de Germán Carrillo) en la consola Python de QGIS:

prefix = "'"
layer = iface.activeLayer()
attr = layer.rendererV2().classAttribute()
attrColor = 'color' # Name of the field to store colors
fieldIndex = layer.dataProvider().fieldNameIndex(attrColor)
attrFeatMap = {}

for cat in layer.rendererV2().categories(): 
  expr = "\""+attr+"\"="+prefix+unicode(cat.value())+prefix
  for f in layer.getFeatures(QgsFeatureRequest(QgsExpression(expr))):
    attrMap = { fieldIndex : cat.symbol().color().name()}
    attrFeatMap[ f.id() ] = attrMap

layer.dataProvider().changeAttributeValues( attrFeatMap ) 

Como se puede ver en la siguiente imagen, el resultado fue completamente satisfactorio (no estoy recibiendo ningún error):

Nota de edición:

Este es el código para Graduated Symbol Renderer:

layer = iface.activeLayer()

provider = layer.dataProvider()

renderer = layer.rendererV2()

feats = [ feat for feat in layer.getFeatures() ]
values = [ feat.attribute('FID')  for feat in feats ] #change for 'NDVI' in case of NDVI field

n = len(feats)

color_feats = [ renderer.symbolForValue(value).color().name() for value in values ]

idx = layer.fieldNameIndex('color')

for i, feat in enumerate(feats):
    new_values = {idx : color_feats[i]}
    provider.changeAttributeValues({i:new_values})

Después de ejecutarlo en la consola Python de QGIS, los valores en el campo 'color' se colocaron como se esperaba.

Utilizando el campo "NDVI" también funciona perfectamente:

Answer 2

OK tengo respuesta

aquí está el plugin hecho por @underdark

y aquí está en GitHub

¡Funciona bien!

Answer 3

Este código debería funcionar para un render graduado:

prefix = "'"
layer = iface.activeLayer()
attr = layer.rendererV2().classAttribute()
attrColor = 'color' # Name of the field to store colors
fieldIndex = layer.dataProvider().fieldNameIndex(attrColor)
attrFeatMap = {}

for cat in layer.rendererV2().ranges():
    expr = "\""+attr+"\">="+prefix+unicode(cat.lowerValue())+prefix+'AND'+"\""+attr+"\"<="+prefix+unicode(cat.upperValue())+prefix
    for f in layer.getFeatures(QgsFeatureRequest(QgsExpression(expr))):
        attrMap = { fieldIndex : cat.symbol().color().name()}
        attrFeatMap[ f.id() ] = attrMap
layer.dataProvider().changeAttributeValues( attrFeatMap ) 

La única diferencia con tu código es el uso de la función ranges() en lugar de categories() y la ligera edición del expr porque esta vez no se trata de un único valor, sino de todo un rango de valores.