How to delete column/field in PyQGIS?

If you get no joy with QGSVectorLayer::deleteAttribute as they docs say: delete an attribute field (but does not commit it) have a look at the data provider. As QGIS deals with lots of feature types with different drivers I suspect that it's left up to the driver as to whether or not a field can be deleted.

Here's some code that I dredged up that may help:

fields = vlayer.dataProvider().fields()

count = 0
for name, field in fields.iteritems():
    if field.name() == fieldName:
        return count
    count += 1

To get the index of the field from the name. In this case I'm not using .upper() but I would recommend it.

After getting the index of the field you (might) delete it using the dataprovider.

fList = list()
fList.append(count)
vlayer.dataProvider().deleteAttributes(fList)

It's expecting a list so the integer needs to be put into a list. I have not used this method; normally I would use ArcMap to delete the field.

As pointed out by RogerHuang, the fields of the layer may need to be updated now that they have been changed to refresh the layers' fields definition:

vlayer.updateFields()

There's several methods but I will mention the ones I normally use, both using the Python Console.


For SINGLE shapefiles:

  • Method 1 - Using a method utilising the QGIS algorithm - Delete Column:

    output_0=processing.runalg('qgis:deletecolumn', myshapefile.shp, "myFirstField", None)
    output_1=processing.runalg('qgis:deletecolumn', output_0['OUTPUT'], "mySecondField", None)
    output_2=processing.runalg('qgis:deletecolumn', output_1['OUTOUT'], "myThirdField", myoutput.shp)
    
  • Method 2 - Deleting the fields directly by selecting the layer from the Layers window (Table of Contents) and using the following code:

    layer = qgis.utils.iface.activeLayer()              # Sets the selected layer as the active layer
    res = layer.dataProvider().deleteAttributes([0])    # The [0] represents the 1st field, so [1] is the 2nd field etc. To delete multiple fields, use the comma to separate. Eg. [0, 2, 4]
    layer.updateFields()                                # Updates the layer with the changes
    

For MULTIPLE shapefiles:

  • Following on from Method 1, you can use the following script which was tested with 2 folders on the Desktop - "Testing" (which contained the shapefiles) and "Results" (which will contain the output shapefiles of the script):

    import processing, os, glob
    
    from os.path import expanduser
    home = expanduser("~")
    
    result_path = home + "\Desktop\Results\\"
    os.chdir(home + "\Desktop\Testing\\")
    
    for fname in glob.glob("*.shp"):
        output_0=processing.runalg('qgis:deletecolumn', fname, "myFirstField", None)
        output_1=processing.runalg('qgis:deletecolumn', output_0['OUTPUT'], "mySecondField", None)
        output_2=processing.runalg('qgis:deletecolumn', output_1['OUTPUT'], "myThirdField", result_path + fname)
    
    print "All done!"
    
  • Following on from Method 2, you can use the following code if you have multiple shapefiles loaded in your Layers window without having to select them individually.

    for layer in QgsMapLayerRegistry.instance().mapLayers().values():
        res = layer.dataProvider().deleteAttributes([0])
        layer.updateFields()
    

Hope this helps!